From 9f5780b6ab32d9ae064c7079333f60557b6f9ed8 Mon Sep 17 00:00:00 2001 From: Frank Fu Date: Wed, 17 Jul 2024 11:30:11 -0400 Subject: [PATCH] very minor argument fix --- src/simsopt/geo/curve.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/simsopt/geo/curve.py b/src/simsopt/geo/curve.py index c5d20587a..e23e682a6 100644 --- 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