Catch more possible dependency errors

typeo23 · Apr 20, 2018 · ced236f · ced236f
1 parent fac1c6b
commit ced236f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/shap/explainers/tree.py b/shap/explainers/tree.py
@@ -6,16 +6,33 @@
     from .. import _cext
 except ImportError:
     pass
+except:
+    print("the C extension is installed...but failed to load!")
+    pass
 
 try:
     import xgboost
 except ImportError:
     pass
+except:
+    print("xgboost is installed...but failed to load!")
+    pass
 
 try:
     import lightgbm
 except ImportError:
     pass
+except:
+    print("lightgbm is installed...but failed to load!")
+    pass
+
+try:
+    import catboost
+except ImportError:
+    pass
+except:
+    print("catboost is installed...but failed to load!")
+    pass
 
 def mapf(t):
     self.tree_shap(t, X[i,:], x_missing, phi[i,:,:])