Do not import PBA if not used

ThibeauWouters · Dec 19, 2024 · 8fe9cb6 · 8fe9cb6
1 parent 6dc0bf1
commit 8fe9cb6
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 83 deletions.
diff --git a/src/.gitignore b/src/.gitignore
@@ -0,0 +1 @@
+fiesta.egg-info
diff --git a/src/fiesta/train/AfterglowData.py b/src/fiesta/train/AfterglowData.py
@@ -9,7 +9,6 @@
 
 from fiesta.constants import days_to_seconds
 import afterglowpy as grb
-from PyBlastAfterglowMag.wrappers import run_grb
 
 
 class AfterglowData:
@@ -372,6 +371,12 @@ def _call_pyblastafterglow(self,
         Args:
             Float[Array, "n_times"]: The flux density in mJys at the given times.
         """
+
+        try:
+            from PyBlastAfterglowMag.wrappers import run_grb
+        except ImportError:
+            raise ImportError("PyBlastAfterglowMag is not installed. Please install it from source")
+
         # Define jet structure (analytic; gaussian) -- 3 free parameters 
         struct = dict(
             struct= self.jet_type, # type of the structure tophat or gaussian

diff --git a/src/fiestaEM.egg-info/PKG-INFO b/src/fiestaEM.egg-info/PKG-INFO
diff --git a/src/fiestaEM.egg-info/SOURCES.txt b/src/fiestaEM.egg-info/SOURCES.txt
diff --git a/src/fiestaEM.egg-info/dependency_links.txt b/src/fiestaEM.egg-info/dependency_links.txt
diff --git a/src/fiestaEM.egg-info/requires.txt b/src/fiestaEM.egg-info/requires.txt
diff --git a/src/fiestaEM.egg-info/top_level.txt b/src/fiestaEM.egg-info/top_level.txt