chore: reinstate repack

SciML · Nov 11, 2024 · 8b26c43 · 8b26c43
1 parent 59c8068
commit 8b26c43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gauss_adjoint.jl b/src/gauss_adjoint.jl
@@ -483,7 +483,7 @@ function vec_pjac!(out, λ, y, t, S::GaussIntegrand)
         copyto!(vec(out), ReverseDiff.deriv(tp))
     elseif sensealg.autojacvec isa ZygoteVJP
         _dy, back = Zygote.pullback(tunables) do tunables
-            vec(f(y, tunables, t))
+            vec(f(y, repack(tunables), t))
         end
         tmp = back(λ)
         if tmp[1] === nothing