examples: Add kwargs to operators and wavesolvers

devitocodes · Aug 18, 2020 · fb16e0a · fb16e0a
1 parent d02ab39
commit fb16e0a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/seismic/elastic/operators.py b/examples/seismic/elastic/operators.py
@@ -63,6 +63,7 @@ def ForwardOperator(model, geometry, space_order=4, save=False, **kwargs):
              model.damp * dt * mu * (grad(v.forward) + grad(v.forward).T))
 
     srcrec = src_rec(v, tau, model, geometry)
-    op = Operator([u_v] + [u_t] + srcrec, subs=model.spacing_map, name="ForwardElastic")
+    op = Operator([u_v] + [u_t] + srcrec, subs=model.spacing_map, name="ForwardElastic",
+                  **kwargs)
     # Substitute spacing terms to reduce flops
     return op
diff --git a/examples/seismic/tti/tti_example.py b/examples/seismic/tti/tti_example.py
@@ -17,7 +17,7 @@ def tti_setup(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0,
     # Source and receiver geometries
     geometry = setup_geometry(model, tn)
 
-    return AnisotropicWaveSolver(model, geometry, space_order=space_order)
+    return AnisotropicWaveSolver(model, geometry, space_order=space_order, **kwargs)
 
 
 def run(shape=(50, 50, 50), spacing=(20.0, 20.0, 20.0), tn=250.0,