import shap only on --shap call, suppress shap's numba deprecation warnings

Author: tschuelia

pypythia/predictor.py

@@ -1,5 +1,5 @@
 import pickle
-import shap
+import warnings
 
 from lightgbm import LGBMRegressor
 from matplotlib.figure import Figure
@@ -112,6 +112,10 @@ def predict(self, query: Dict) -> float:
         return prediction
 
     def plot_shapley_values(self, query: Dict) -> Figure:
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore")
+            import shap
+
         if isinstance(self.predictor, RandomForestRegressor):
             raise PyPythiaException("Cannot infer shapley values for scikit-learn predictors")
 

setup.cfg

@@ -6,7 +6,7 @@ long_description_content_type = text/markdown
 url = https://github.com/tschuelia/PyPythia
 author = Julia Haag
 author_email = [email protected]
-version = 1.1.0
+version = 1.1.2
 classifiers =
     Programming Language :: Python :: 3.7
     Programming Language :: Python :: 3.8