Made pyrr into an optional dependency.

hpparvi · Nov 26, 2024 · 0946559 · 0946559
1 parent 64d3ae3
commit 0946559
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/pytransit/__init__.py b/pytransit/__init__.py
@@ -41,7 +41,7 @@
 
 """
 
-__version__ = '2.6.12'
+__version__ = '2.6.13'
 
 # Generic
 # -------

diff --git a/pytransit/utils/octasphere.py b/pytransit/utils/octasphere.py
@@ -21,8 +21,12 @@
 
 from math import sin, cos, acos, pi
 from numpy import empty, array, vstack, cross, dot
-from pyrr import quaternion
 
+try:
+    from pyrr import quaternion
+    with_pyrr = True
+except ImportError:
+    with_pyrr = False
 
 def octasphere(ndivisions: int):
     """Creates a unit sphere using octagon subdivision.
@@ -31,6 +35,9 @@ def octasphere(ndivisions: int):
     by Philip Rideout (https://prideout.net/blog/octasphere).
     """
 
+    if not with_pyrr:
+        raise ModuleNotFoundError("The pyrr package is required for GDModel visualization.")
+
     n = 2**ndivisions + 1
     num_verts = n * (n + 1) // 2
     verts = empty((num_verts, 3))

diff --git a/requirements.txt b/requirements.txt
@@ -17,5 +17,4 @@ emcee
 pyopencl
 corner
 celerite
-pyrr
 meepmeep