adding logic for when matplotlib isn't available

SuperCowPowers · Sep 3, 2024 · a525c5d · a525c5d
1 parent 59400ab
commit a525c5d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/sageworks/repl/sageworks_shell.py b/src/sageworks/repl/sageworks_shell.py
@@ -11,8 +11,9 @@
 
 try:
     import matplotlib.pyplot as plt  # noqa
+    HAVE_MATPLOTLIB = True
 except ImportError:
-    pass
+    HAVE_MATPLOTLIB = False
 
 
 # SageWorks Imports
@@ -89,7 +90,8 @@ def __init__(self):
         self.shell = InteractiveShellEmbed(config=config, banner1="", exit_msg="Goodbye from SageWorks!")
 
         # Set matplotlib to interactive mode
-        self.shell.run_line_magic("matplotlib", "auto")
+        if HAVE_MATPLOTLIB:
+            self.shell.run_line_magic("matplotlib", "auto")
 
         # Register our custom commands
         self.commands["help"] = self.help