fix: make diabatization.driver importable w/o pysisyphus_addons

eljost · Nov 14, 2024 · adc9c8f · adc9c8f
1 parent 00f1d7e
commit adc9c8f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pysisyphus/diabatization/driver.py b/pysisyphus/diabatization/driver.py
@@ -34,9 +34,15 @@
     norm_ci_coeffs,
 )
 
-# pysisyphus_addons imports below
-from pysisyphus_addons.diabatization import intor
-from pysisyphus_addons.grid import grid
+
+try:
+    # pysisyphus_addons imports below
+    from pysisyphus_addons.diabatization import intor
+    from pysisyphus_addons.grid import grid
+
+    can_pysis_addons = True
+except (ImportError, ModuleNotFoundError):
+    can_pysis_addons = False
 
 
 class DiaKind(enum.Flag):