yq 3 месяцев назад
Родитель
Сommit
d80ca9b21d
1 измененных файлов с 26 добавлено и 3 удалено
  1. 26 3
      init/__init__.py

+ 26 - 3
init/__init__.py

@@ -4,6 +4,7 @@
 @time: 2024/10/31
 @desc: 一些资源初始化
 """
+import sys
 
 import matplotlib
 
@@ -21,11 +22,33 @@ def init():
     plt.rcParams['figure.max_open_warning'] = 1000
 
 
+def _custom_warning_filter(action, message, category, module, filename, line):
+    # 如果警告消息包含“ignore this warning”,则忽略该警告
+    if "matplotlib.font_manager" in str(message):
+        return False
+    # 否则,显示警告
+    return True
+
+
 def warning_ignore():
-    from pandas.core.common import SettingWithCopyWarning
     import warnings
-    warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
-    warnings.simplefilter(action="ignore", category=RuntimeWarning)
+
+    # warnings.simplefilter(action="ignore", category=RuntimeWarning)
+    # warnings.simplefilter(action="ignore", category=UserWarning)
+    warnings.simplefilter(action="ignore", category=FutureWarning)
+
+    warnings.filterwarnings(action="ignore", module="matplotlib")
+    warnings.filterwarnings(action="ignore", module="dataframe_image")
+    warnings.filterwarnings(action="ignore", module="pandas")
+    warnings.filterwarnings(action="ignore", module="scorecardpy")
+
+    if "3.6" in sys.version:
+        from pandas.core.common import SettingWithCopyWarning
+        warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
+
+    if "3.10" in sys.version:
+        from pandas.errors import SettingWithCopyWarning
+        warnings.simplefilter(action="ignore", category=SettingWithCopyWarning)
 
 
 if __name__ == "__main__":