very minor argument fix

hiddenSymmetries · Jul 17, 2024 · 9f5780b · 9f5780b
1 parent 431e3d6
commit 9f5780b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/simsopt/geo/curve.py b/src/simsopt/geo/curve.py
@@ -1519,8 +1519,8 @@ def dkappa_by_dcoeff_vjp(self, v):
 
         """
         return Derivative({
-            self: self.dkappa_by_dcoeff_vjp_jax(self.get_dofs(), v),
-            self.surf: self.dkappa_by_dsurf_vjp_jax(self.surf.get_dofs(), v)
+            self: self.dkappa_by_dcoeff_vjp_jax(self.get_dofs(), self.surf.get_dofs(), v),
+            self.surf: self.dkappa_by_dsurf_vjp_jax(self.get_dofs(), self.surf.get_dofs(), v)
             })
 
     # TORSION