Merge pull request #58 from agencyenterprise/fix-extra-kwarg

Pop kwarg to avoid `TypeError`
trustimaging · Oct 12, 2023 · 31d52cf · 31d52cf
2 parents fb47928 + 9301fd3
commit 31d52cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stride/physics/iso_acoustic/devito.py b/stride/physics/iso_acoustic/devito.py
@@ -259,7 +259,7 @@ async def before_forward(self, wavelets, vp, rho=None, alpha=None, **kwargs):
         num_sources = shot.num_points_sources
         num_receivers = shot.num_points_receivers
 
-        save_wavefield = kwargs.get('save_wavefield', False)
+        save_wavefield = kwargs.pop('save_wavefield', False)
         if save_wavefield is False:
             save_wavefield = vp.needs_grad
             if rho is not None: