Added import error for scipy Rotation

lab-cosmo · Aug 28, 2023 · 8c87fcc · 8c87fcc
1 parent 6059ae2
commit 8c87fcc
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/chemiscope/structures/_shapes.py b/python/chemiscope/structures/_shapes.py
@@ -146,8 +146,12 @@ def ellipsoid_from_tensor(tensor, scale=1.0, force_positive=False):
     rotation = eigenvectors.T
 
     # converts to quaternion
-    from scipy.spatial.transform import Rotation
-
+    try:
+        from scipy.spatial.transform import Rotation
+    except ImportError as e:
+        raise RuntimeError(
+            "scipy is required to construct arrow shapes"
+        ) from e
     quaternion = Rotation.from_matrix(rotation).as_quat()
 
     return dict(