From 9301fd301f1d388a090a895d25734fcae85bbb09 Mon Sep 17 00:00:00 2001 From: Charles Guan Date: Tue, 10 Oct 2023 10:00:24 -0700 Subject: [PATCH] Pop kwarg to avoid TypeError _stencil() got multiple values for keyword argument 'save_wavefield' --- stride/physics/iso_acoustic/devito.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stride/physics/iso_acoustic/devito.py b/stride/physics/iso_acoustic/devito.py index 68f30aea..feeb8d63 100644 --- 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: