Merge pull request #170 from ChevronETC/fixlazy

fixup no layers
ChevronETC · Nov 17, 2023 · 06ba9b6 · 06ba9b6
2 parents 9ae5ac8 + a625b30
commit 06ba9b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Devito.jl b/src/Devito.jl
@@ -882,10 +882,10 @@ function TimeFunction(args...; allowpro=true, lazy=true, kwargs...)
             # TODO: Generate MFE and submit as issue to PyCall
             py"""
             import devitopro
-            def serializedtimefunc(serialization, **kwargs):
+            def serializedtimefunc(**kwargs):
                 return devitopro.TimeFunction(layers=devitopro.types.enriched.Disk, **kwargs)
             """
-            o = py"serializedtimefunc"(serialization; Devito.reversedims(kwargs)...)
+            o = py"serializedtimefunc"(; Devito.reversedims(kwargs)...)
         end
     else
         o = pycall(devito.TimeFunction, PyObject, args...; reversedims(kwargs)...)