Run the loop in parallel

tardis-sn · Jul 30, 2024 · 1a47b9b · 1a47b9b
1 parent 7231707
commit 1a47b9b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tardis/transport/montecarlo/montecarlo_main_loop.py b/tardis/transport/montecarlo/montecarlo_main_loop.py
@@ -179,8 +179,8 @@ def montecarlo_main_loop(
         )
 
     if montecarlo_globals.ENABLE_RPACKET_TRACKING:
-        for rpacket_tracker in rpacket_trackers:
-            rpacket_tracker.finalize_array()
+        for i in prange(no_of_packets):
+            rpacket_trackers[i].finalize_array()
 
     return (
         v_packets_energy_hist,