Updated the shgrav.py package so that it can be activated during docu…

…mentation building
MintonGroup · Mar 1, 2024 · 7d48edb · 7d48edb
1 parent dc8511b
commit 7d48edb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/.readthedocs.yaml b/.readthedocs.yaml
@@ -20,7 +20,6 @@ build:
   jobs:
     pre_build:
       - ford fortran_docs.md 
-      - pip install pyshtools
       - DOC_BUILD=1 pip install .
 python:
   install:

diff --git a/swiftest/shgrav.py b/swiftest/shgrav.py
@@ -14,13 +14,20 @@
 # 
 
 from .constants import GC
+import os
 
-try:
-    import pyshtools as pysh
+PYSHTOOLS_AVAILABLE = False
+
+if 'READTHEDOCS' in os.environ:
+    # Assume pyshtools is available when building on ReadTheDocs
     PYSHTOOLS_AVAILABLE = True
-except ModuleNotFoundError:
-    PYSHTOOLS_AVAILABLE = False
-    print("pyshtools is not installed. Some features will be unavailable.")
+else:
+    try:
+        import pyshtools as pysh
+        PYSHTOOLS_AVAILABLE = True
+    except ModuleNotFoundError:
+        print("pyshtools is not installed. Some features will be unavailable.")
+
 
 if PYSHTOOLS_AVAILABLE: