update classes

python-hydro · Oct 18, 2024 · cdea66d · cdea66d
1 parent f5d6a84
commit cdea66d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/pyro/compressible_fv4/simulation.py b/pyro/compressible_fv4/simulation.py
@@ -6,9 +6,11 @@
 class Simulation(compressible_rk.Simulation):
 
     def __init__(self, solver_name, problem_name, problem_func, rp, *,
-                 problem_finalize_func=None, timers=None, data_class=fv.FV2d):
+                 problem_finalize_func=None, problem_source_func=None,
+                 timers=None, data_class=fv.FV2d):
         super().__init__(solver_name, problem_name, problem_func, rp,
                          problem_finalize_func=problem_finalize_func,
+                         problem_sourcE_func=problem_source_func,
                          timers=timers, data_class=data_class)
 
     def substep(self, myd):

diff --git a/pyro/lm_atm/simulation.py b/pyro/lm_atm/simulation.py
@@ -36,10 +36,13 @@ def jp(self, shift, buf=0):
 class Simulation(NullSimulation):
 
     def __init__(self, solver_name, problem_name, problem_func, rp, *,
-                 problem_finalize_func=None, timers=None):
+                 problem_finalize_func=None, problem_source_func=None,
+                 timers=None):
 
         super().__init__(solver_name, problem_name, problem_func, rp,
-                         problem_finalize_func=problem_finalize_func, timers=timers)
+                         problem_finalize_func=problem_finalize_func,
+                         problem_source_func=problem_source_func,
+                         timers=timers)
 
         self.base = {}
         self.aux_data = None