fix explanation dashboard failing to load and serialization error on …

…notebooks (#153)
interpretml · Jan 21, 2020 · 7d2a673 · 7d2a673
1 parent d5c3365
commit 7d2a673
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/interpret_community/widget/explanation_dashboard.py b/python/interpret_community/widget/explanation_dashboard.py
@@ -7,7 +7,7 @@
 import os
 import json
 from .explanation_dashboard_input import ExplanationDashboardInput
-from ._internal import DatabricksInterfaceConstants
+from ._internal.constants import DatabricksInterfaceConstants
 
 """Explanation Dashboard Class.
 

diff --git a/python/interpret_community/widget/explanation_dashboard_input.py b/python/interpret_community/widget/explanation_dashboard_input.py
@@ -96,6 +96,9 @@ class and for the regression case a method of predict() returning the prediction
             try:
                 local_explanation["scores"] = self._convert_to_list(local_explanation["scores"])
                 local_explanation["intercept"] = self._convert_to_list(local_explanation["intercept"])
+                # We can ignore perf explanation data.  Note if it is added back at any point,
+                # the numpy values will need to be converted to python, otherwise serialization fails.
+                local_explanation["perf"] = None
                 self.dashboard_input[ExplanationDashboardInterface.LOCAL_EXPLANATIONS] = local_explanation
             except Exception:
                 raise ValueError("Unsupported local explanation type")