restructure plotting import to be testable

SBC-Utrecht · Feb 27, 2024 · 9c71ff7 · 9c71ff7
1 parent f4dd1fa
commit 9c71ff7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/pytom_tm/plotting.py b/src/pytom_tm/plotting.py
@@ -5,13 +5,13 @@
 from scipy.optimize import curve_fit
 from scipy.special import erf
 
-if find_spec("matplotlib") is None or find_spec("seaborn") is None:
+try:
+    import matplotlib.pyplot as plt
+    import seaborn as sns
+except ModuleNotFoundError:
     raise RuntimeError(
         "ROC estimation can only be done when matplotlib and seaborn are installed."
     )
-else:
-    import matplotlib.pyplot as plt
-    import seaborn as sns
 sns.set(context="talk", style="ticks")