Change parameters in benchmarks (#2782)

* change parameters in benchmarks Signed-off-by: Asish Kumar <[email protected]> * change parameters in benchmarks Signed-off-by: Asish Kumar <[email protected]> --------- Signed-off-by: Asish Kumar <[email protected]>
tardis-sn · Aug 12, 2024 · 18dcec8 · 18dcec8
1 parent 31fa786
commit 18dcec8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 12 deletions.
diff --git a/benchmarks/opacities_opacity.py b/benchmarks/opacities_opacity.py
@@ -28,7 +28,7 @@ def time_photoabsorption_opacity_calculation(self):
         )
 
     def time_pair_creation_opacity_calculation(self):
-        energy = 255.9
+        energy = 1240
         ejecta_density = 100000.009
         iron_group_fraction = 0.5
         calculate_opacity.pair_creation_opacity_calculation(

diff --git a/benchmarks/transport_montecarlo_packet_trackers.py b/benchmarks/transport_montecarlo_packet_trackers.py
@@ -9,7 +9,6 @@
 from tardis.transport.montecarlo import packet_trackers
 
 
-@parameterize({"num_packets": [10, 100], "length": [10, 50]})
 class BenchmarkTransportMontecarloPacketTrackers(BenchmarkBase):
     """
     Class to benchmark the numba R packet function.
@@ -18,21 +17,17 @@ class BenchmarkTransportMontecarloPacketTrackers(BenchmarkBase):
     repeat = 2
 
     @functools.cache
-    def setup(self, num_packets, length):
+    def setup(self):
         sim = self.simulation_rpacket_tracking_enabled
         self.TransportState = sim.transport.transport_state
 
-    def time_rpacket_trackers_to_dataframe(self, num_packets, length):
+    def time_rpacket_trackers_to_dataframe(self):
         packet_trackers.rpacket_trackers_to_dataframe(
             self.TransportState.rpacket_tracker
         )
 
-    def time_generate_rpacket_tracker_list(self, num_packets, length):
-        packet_trackers.generate_rpacket_tracker_list(num_packets, length)
+    def time_generate_rpacket_tracker_list(self):
+        packet_trackers.generate_rpacket_tracker_list(50, 10)
 
-    def time_generate_rpacket_last_interaction_tracker_list(
-        self, num_packets, length
-    ):
-        packet_trackers.generate_rpacket_last_interaction_tracker_list(
-            num_packets
-        )
+    def time_generate_rpacket_last_interaction_tracker_list(self):
+        packet_trackers.generate_rpacket_last_interaction_tracker_list(50)