Fix setvjp for SteadyStateAdjoint

SciML · Apr 14, 2023 · ed76eb3 · ed76eb3
1 parent 2aa6a75
commit ed76eb3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sensitivity_algorithms.jl b/src/sensitivity_algorithms.jl
@@ -993,7 +993,8 @@ Base.@pure function SteadyStateAdjoint(; chunk_size = 0, autodiff = true,
     SteadyStateAdjoint{chunk_size, autodiff, diff_type, typeof(autojacvec), typeof(linsolve)
                        }(autojacvec, linsolve)
 end
-function setvjp(sensealg::SteadyStateAdjoint{CS, AD, FDT, LS}, vjp) where {CS, AD, FDT, LS}
+function setvjp(sensealg::SteadyStateAdjoint{CS, AD, FDT, VJP, LS},
+                vjp) where {CS, AD, FDT, VJP, LS}
     SteadyStateAdjoint{CS, AD, FDT, typeof(vjp), LS}(vjp, sensealg.linsolve)
 end