Skip to content

Commit e80bb70

Browse files
committed
Fix type stability in getadjointsensitivities
1 parent f0aa710 commit e80bb70

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/Simulation.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ import SciMLBase: AbstractODESolution, HermiteInterpolation, AbstractDiffEqInter
33
using SciMLSensitivity
44
using ForwardDiff
55
using PreallocationTools
6+
using LinearAlgebra
67

78
abstract type AbstractSimulation end
89
export AbstractSimulation
@@ -561,7 +562,7 @@ function getadjointsensitivities(bsol::Simulation, target::String, solver; sensa
561562
dpadj ./= bsol.sol(bsol.sol.t[end])[ind]
562563
end
563564
end
564-
return dpadj
565+
return dpadj::LinearAlgebra.Adjoint{Float64, Vector{Float64}}
565566
end
566567

567568
function getadjointsensitivities(syssim::Q, bsol::W3, target::String, solver::W; sensalg::W2=InterpolatingAdjoint(autojacvec=ReverseDiffVJP(false)),
@@ -605,7 +606,7 @@ function getadjointsensitivities(syssim::Q, bsol::W3, target::String, solver::W;
605606
dpadj ./= bsol.sol(bsol.sol.t[end])[ind]
606607
end
607608
end
608-
return dpadj
609+
return dpadj::LinearAlgebra.Adjoint{Float64, Vector{Float64}}
609610
end
610611
export getadjointsensitivities
611612

0 commit comments

Comments
 (0)