Fix for numba_config import

tardis-sn · Apr 21, 2021 · f9c2645 · f9c2645
1 parent cf853e9
commit f9c2645
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tardis/montecarlo/montecarlo_numba/estimators.py b/tardis/montecarlo/montecarlo_numba/estimators.py
@@ -9,6 +9,10 @@
     calc_packet_energy_full_relativity,
 )
 
+from tardis.montecarlo.montecarlo_numba.numba_config import (
+    ENABLE_FULL_RELATIVITY,
+)
+
 
 @njit(**njit_dict_no_parallel)
 def set_estimators(r_packet, distance, numba_estimator, comov_nu, comov_energy):
@@ -59,7 +63,7 @@ def update_line_estimators(
     ( time_explosion * C)
     """
 
-    if not numba_config.ENABLE_FULL_RELATIVITY:
+    if not ENABLE_FULL_RELATIVITY:
         energy = calc_packet_energy(r_packet, distance_trace, time_explosion)
     else:
         energy = calc_packet_energy_full_relativity(r_packet)