Clean-up of environment code.

interpretml · Sep 18, 2019 · 4e47d45 · 4e47d45
1 parent ef9caaa
commit 4e47d45
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/python/interpret-core/interpret/provider/visualize.py b/python/interpret-core/interpret/provider/visualize.py
@@ -39,11 +39,7 @@ def render(self, explanation, key=-1, **kwargs):
 
         # Output to front-end/file
         self._preserve_output(
-            explanation.name,
-            visual,
-            selector_key=key,
-            file_name=file_name,
-            **kwargs
+            explanation.name, visual, selector_key=key, file_name=file_name, **kwargs
         )
         return None
 

diff --git a/python/interpret-core/interpret/utils/environment.py b/python/interpret-core/interpret/utils/environment.py
@@ -38,11 +38,9 @@ def _detect_ipython_zmq():
     Returns:
         True if called in IPython notebook or qtconsole.
     """
-
-    if _detect_ipython() is False:
-        return False
-
     try:
+        from IPython import get_ipython
+
         shell = get_ipython().__class__.__name__
         if shell == "ZMQInteractiveShell":
             return True  # Jupyter notebook or qtconsole