Ver Fonte

modify: jupyter输出优化

yq há 3 meses atrás
pai
commit
393ca10eaf
2 ficheiros alterados com 7 adições e 5 exclusões
  1. 7 0
      feature/filter_strategy_base.py
  2. 0 5
      init/__init__.py

+ 7 - 0
feature/filter_strategy_base.py

@@ -18,6 +18,13 @@ class FilterStrategyBase(metaclass=abc.ABCMeta):
         else:
             self._data_process_config = DataProcessConfigEntity(*args, **kwargs)
 
+        jupyter = self._data_process_config.jupyter
+        if jupyter:
+            from pandas.core.common import SettingWithCopyWarning
+            import warnings
+            warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
+            warnings.simplefilter(action="ignore", category=RuntimeWarning)
+
     @property
     def data_process_config(self):
         return self._data_process_config

+ 0 - 5
init/__init__.py

@@ -5,16 +5,11 @@
 @desc: 一些资源初始化
 """
 
-import warnings
-
 import matplotlib
 
 matplotlib.use('Agg')
 
 import matplotlib.pyplot as plt
-from pandas.core.common import SettingWithCopyWarning
-
-warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
 
 __all__ = ['init']