fix missing argument dr problem

PlasmaControl · Oct 7, 2024 · 891c914 · 891c914
1 parent 9631353
commit 891c914
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/desc/basis.py b/desc/basis.py
@@ -1509,7 +1509,7 @@ def custom_jvp_with_jit(func):
         custom_jvp,
         nondiff_argnums=(3,),
     )
-    def dummy(r, l, m, dr):
+    def dummy(r, l, m, dr=0):
         return func(r, l, m, dr)
 
     @dummy.defjvp