Merge pull request #698 from Sichao25/root_folder

Debug set_figure_params
aristoteleo · Aug 5, 2024 · 53a17b9 · 53a17b9
2 parents 117ceb5 + 17bff01
commit 53a17b9
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/dynamo/configuration.py b/dynamo/configuration.py
@@ -789,14 +789,17 @@ def set_figure_params(
             for more details.
     """
 
-    try:
-        import IPython
-
-        if isinstance(ipython_format, str):
-            ipython_format = [ipython_format]
-        IPython.display.set_matplotlib_formats(*ipython_format)
-    except Exception:
-        pass
+    import builtins
+
+    if getattr(builtins, "__IPYTHON__", False):
+        try:
+            import matplotlib_inline.backend_inline
+
+            if isinstance(ipython_format, str):
+                ipython_format = [ipython_format]
+            matplotlib_inline.backend_inline.set_matplotlib_formats(*ipython_format)
+        except Exception:
+            pass
 
     from matplotlib import rcParams