cache in setup function

Signed-off-by: Asish Kumar <[email protected]>

benchmarks/transport_montecarlo_interaction.py

@@ -2,18 +2,20 @@
 Basic TARDIS Benchmark.
 """
 
+import functools
+
 import tardis.transport.montecarlo.interaction as interaction
 from benchmarks.benchmark_base import BenchmarkBase
 
 
-# check it
 class BenchmarkTransportMontecarloInteraction(BenchmarkBase):
     """
     Class to benchmark the numba interaction function.
     """
 
-    repeat = 3
+    repeat = 3 
 
+    @functools.cache
     def setup(self):
         self.Packet = self.packet
         self.time_explosion = self.verysimple_time_explosion

benchmarks/transport_montecarlo_main_loop.py

@@ -2,6 +2,8 @@
 Basic TARDIS Benchmark.
 """
 
+import functools
+
 from benchmarks.benchmark_base import BenchmarkBase
 from tardis.transport.montecarlo.montecarlo_main_loop import (
     montecarlo_main_loop,
@@ -15,6 +17,7 @@ class to benchmark montecarlo_main_loop function.
 
     repeat = 3
 
+    @functools.cache
     def setup(self):
         self.packet_collection = self.transport_state.packet_collection
         self.geometry_state = self.transport_state.geometry_state

benchmarks/transport_montecarlo_packet_trackers.py

@@ -1,6 +1,8 @@
 """
 Basic TARDIS Benchmark.
 """
+import functools
+
 from asv_runner.benchmarks.mark import parameterize
 
 from benchmarks.benchmark_base import BenchmarkBase
@@ -15,6 +17,7 @@ class BenchmarkTransportMontecarloPacketTrackers(BenchmarkBase):
 
     repeat = 2
 
+    @functools.cache
     def setup(self, num_packets, length):
         sim = self.simulation_rpacket_tracking_enabled
         self.TransportState = sim.transport.transport_state