pims_wrapper: fix matplotlib import

lens-biophotonics · Jun 1, 2022 · 7423e6e · 7423e6e
1 parent 4909655
commit 7423e6e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/zetastitcher/io/pims_wrapper.py b/zetastitcher/io/pims_wrapper.py
@@ -1,7 +1,10 @@
-# Suppress warnings when importing pims on a machine without display server
-# The warnings are due to import matplotlib.pyplot as plt in pim's display.py
-import matplotlib as mpl
-mpl.use('Agg')
+try:
+    # Suppress warnings when importing pims on a machine without display server
+    # The warnings are due to import matplotlib.pyplot as plt in pim's display.py
+    import matplotlib as mpl
+    mpl.use('Agg')
+except ModuleNotFoundError:
+    pass
 
 from pathlib import Path
 from pims import ImageSequence