diff --git a/examples/wave/wave-eager-mpi.py b/examples/wave/wave-eager-mpi.py index 4b408bb599c5002a32325f5befd9ab66ab02cee9..98e020df9b12b76ad757676e1ab3850e57aebf2d 100644 --- a/examples/wave/wave-eager-mpi.py +++ b/examples/wave/wave-eager-mpi.py @@ -106,7 +106,7 @@ def rk4_step(y, t, h, f): return y + h/6*(k1 + 2*k2 + 2*k3 + k4) -def bump(discr, actx, t=0): +def bump(actx, discr, t=0): source_center = np.array([0.2, 0.35, 0.1])[:discr.dim] source_width = 0.05 source_omega = 3 @@ -171,7 +171,7 @@ def main(): raise ValueError("don't have a stable time step guesstimate") fields = flat_obj_array( - bump(discr, actx), + bump(actx, discr), [discr.zeros(actx) for i in range(discr.dim)] ) diff --git a/examples/wave/wave-eager.py b/examples/wave/wave-eager.py index 7450f435798c249f97dd11e03cd59dbd1e95b542..d9d57eacf7a454148279bea5eaf2162d577c2440 100644 --- a/examples/wave/wave-eager.py +++ b/examples/wave/wave-eager.py @@ -100,7 +100,7 @@ def rk4_step(y, t, h, f): return y + h/6*(k1 + 2*k2 + 2*k3 + k4) -def bump(discr, actx, t=0): +def bump(actx, discr, t=0): source_center = np.array([0.2, 0.35, 0.1])[:discr.dim] source_width = 0.05 source_omega = 3 @@ -147,7 +147,7 @@ def main(): discr = EagerDGDiscretization(actx, mesh, order=order) fields = flat_obj_array( - bump(discr, actx), + bump(actx, discr), [discr.zeros(actx) for i in range(discr.dim)] )