Merge pull request #100 from jlazar17/main

Add res_dt as __init__ kwarg
icecube · Jan 5, 2024 · b80db23 · b80db23
2 parents 2aa4b68 + 12f98c9
commit b80db23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/asteria/simulation.py b/python/asteria/simulation.py
@@ -27,7 +27,7 @@
 class Simulation:
     """ Top-level class for performing ASTERIA's core simulation routine, and handler for the resulting outputs
     """
-    def __init__(self, config=None, *, model=None, distance=10 * u.kpc, flavors=None, hierarchy=None,
+    def __init__(self, config=None, *, model=None, distance=10 * u.kpc, res_dt=2 * u.ms, flavors=None, hierarchy=None,
                  interactions=Interactions, mixing_scheme=None, mixing_angle=None, E=None, Emin=None, Emax=None,
                  dE=None, t=None, tmin=None, tmax=None, dt=None, geomfile=None, effvolfile=None):
         self.param = {}
@@ -64,7 +64,7 @@ def __init__(self, config=None, *, model=None, distance=10 * u.kpc, flavors=None
             self.energy = E
             self.time = t
             self._sim_dt = _dt
-            self._res_dt = 2 * u.ms  # TODO: Add config/arg option for this
+            self._res_dt = res_dt
             self._res_offset = 0 * u.s  # TODO: Add config/arg option for this
             if flavors is None:
                 self.flavors = Flavor