backcompat imports to work with older versions of interpret-core (#574)

interpretml · Aug 22, 2023 · 658fd84 · 658fd84
1 parent b70d695
commit 658fd84
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/python/interpret_community/explanation/explanation.py b/python/interpret_community/explanation/explanation.py
@@ -10,9 +10,17 @@
 
 import numpy as np
 import pandas as pd
-from interpret.utils._explanation import (gen_global_selector,
-                                          gen_local_selector,
-                                          gen_name_from_class)
+
+try:
+    from interpret.utils._explanation import (gen_global_selector,
+                                              gen_local_selector,
+                                              gen_name_from_class)
+except ImportError:
+    # backcompat with older versions of interpret-core
+    from interpret.utils import (gen_global_selector,
+                                 gen_local_selector,
+                                 gen_name_from_class)
+
 from ml_wrappers import DatasetWrapper
 from scipy.sparse import issparse