Move Band_Eflux import inside get_integrated_spectral_model

cositools · Dec 18, 2024 · c88a563 · c88a563
1 parent 1d22d95
commit c88a563
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cosipy/response/functions.py b/cosipy/response/functions.py
@@ -8,8 +8,6 @@
 
 from threeML import Band, DiracDelta, Constant, Line, Quadratic, Cubic, Quartic, StepFunction, StepFunctionUpper, Cosine_Prior, Uniform_prior, PhAbs, Gaussian
 
-from cosipy.threeml import Band_Eflux
-
 def get_integrated_spectral_model(spectrum, energy_axis):
     """
     Get the photon fluxes integrated over given energy bins with an input astropy spectral model
@@ -37,6 +35,8 @@ def get_integrated_spectral_model(spectrum, energy_axis):
     over each energy bin, and returns the result as a Histogram object.
     """
 
+    from cosipy.threeml import Band_Eflux
+
     spectrum_unit = None
 
     for item in spectrum.parameters: