diff --git a/drexml/cli/cli.py b/drexml/cli/cli.py index d9cfae7d..65f3c6d4 100644 --- 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 index 773ef8a3..9a4324c0 100644 --- 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