Update benchmark

tardis-sn · Jul 27, 2024 · 0c1c35a · 0c1c35a
1 parent f04bb84
commit 0c1c35a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/benchmarks/benchmark_base.py b/benchmarks/benchmark_base.py
@@ -30,7 +30,6 @@
 )
 from tardis.transport.montecarlo.packet_trackers import (
     RPacketTracker,
-    RPacketLastInteractionTracker,
     generate_rpacket_last_interaction_tracker,
 )
 
@@ -420,6 +419,6 @@ def estimators(self):
         )
 
     @property
-    def rpacket_tracker(self):
+    def rpacket_tracker_list(self):
         no_of_packets = len(self.transport_state.packet_collection)
         return generate_rpacket_last_interaction_tracker_list(no_of_packets)
diff --git a/benchmarks/transport_montecarlo_main_loop.py b/benchmarks/transport_montecarlo_main_loop.py
@@ -22,7 +22,7 @@ def time_montecarlo_main_loop(self):
             self.montecarlo_configuration,
             self.transport_state.radfield_mc_estimators,
             self.nb_simulation_verysimple.transport.spectrum_frequency_grid.value,
-            self.rpacket_tracker,
+            self.rpacket_tracker_list,
             self.montecarlo_configuration.NUMBER_OF_VPACKETS,
             iteration=0,
             show_progress_bars=False,