Ignore numba warnings in cli.

loucerac · May 15, 2023 · a9b0d13 · a9b0d13
1 parent 27b3090
commit a9b0d13
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/drexml/cli/cli.py b/drexml/cli/cli.py
@@ -14,9 +14,20 @@
 import shutil
 import subprocess
 import sys
+import warnings
 
 import click
 import joblib
+from numba.core.errors import NumbaDeprecationWarning, NumbaPendingDeprecationWarning
+
+with warnings.catch_warnings():
+    warnings.filterwarnings(
+        "ignore", module="shap", message="IPython could not be loaded!"
+    )
+    warnings.filterwarnings("ignore", module="shap", category=NumbaDeprecationWarning)
+    warnings.filterwarnings(
+        "ignore", module="shap", category=NumbaPendingDeprecationWarning
+    )
 
 from drexml.plotting import plot_metrics
 from drexml.utils import (

diff --git a/drexml/utils.py b/drexml/utils.py
@@ -11,11 +11,16 @@
 
 import joblib
 import pandas as pd
+from numba.core.errors import NumbaDeprecationWarning, NumbaPendingDeprecationWarning
 
 with warnings.catch_warnings():
     warnings.filterwarnings(
         "ignore", module="shap", message="IPython could not be loaded!"
     )
+    warnings.filterwarnings("ignore", module="shap", category=NumbaDeprecationWarning)
+    warnings.filterwarnings(
+        "ignore", module="shap", category=NumbaPendingDeprecationWarning
+    )
     import shap
 
 from sklearn.model_selection import ShuffleSplit, train_test_split