Fix Adjoint Sensitivities

src/Simulation.jl

@@ -3,6 +3,7 @@
 using SciMLSensitivity
 using ForwardDiff
 using PreallocationTools
+using LinearAlgebra
 
 abstract type AbstractSimulation end
 export AbstractSimulation
@@ -319,7 +320,7 @@
 function rops(ssys::SystemSimulation, name, t)
     domains = getfield.(ssys.sims, :domain)
     ind = findfirst(isequal(name), ssys.names)
     Nrxns = sum([length(sim.domain.phase.reactions) for sim in ssys.sims]) + sum(Vector{Int}([length(inter.reactions) for inter in ssys.interfaces if hasproperty(inter, :reactions)]))
     Nspcs = sum([length(getphasespecies(sim.domain.phase)) for sim in ssys.sims])
     cstot = zeros(Nspcs)
     vns = Array{Any,1}(undef, length(domains))
@@ -546,10 +547,10 @@
         end
     else
         if target in ["T", "V", "P", "mass"] || !isempty(bsol.interfaces)
             du0, dpadj = adjoint_sensitivities(bsol.sol, solver; g=g, dgdu_continuous=gdurevdiff, 
                     dgdp_continuous=dgdprevdiff, sensealg=sensalg, abstol=abstol, reltol=reltol, kwargs...)
         else
             du0, dpadj = adjoint_sensitivities(bsol.sol, solver; g=sensg, dgdu_continuous=dsensgdurevdiff,
                      dgdp_continuous=dsensgdprevdiff, sensealg=sensalg, abstol=abstol, reltol=reltol, kwargs...)
         end
     end
@@ -561,7 +562,7 @@
             dpadj ./= bsol.sol(bsol.sol.t[end])[ind]
         end
     end
-    return dpadj
+    return dpadj::LinearAlgebra.Adjoint{Float64, Vector{Float64}}
 end
 
 function getadjointsensitivities(syssim::Q, bsol::W3, target::String, solver::W; sensalg::W2=InterpolatingAdjoint(autojacvec=ReverseDiffVJP(false)),
@@ -605,7 +606,7 @@
             dpadj ./= bsol.sol(bsol.sol.t[end])[ind]
         end
     end
-    return dpadj
+    return dpadj::LinearAlgebra.Adjoint{Float64, Vector{Float64}}
 end
 export getadjointsensitivities
 
@@ -782,15 +783,15 @@
     CSV.write(save_name, df)
 end
 
 function save(syss::T, save_name::String) where {T<:SystemSimulation}
     df = DataFrame(syss.sol)
     for sim in syss.sims
         rename!(df, names(df)[sim.domain.indexes[1]:sim.domain.indexes[2]] .=> sim.names .* "($(sim.domain.phase.name))")
         for (thermovariable, index) in sim.domain.thermovariabledict
             rename!(df, names(df)[index] => thermovariable)
         end
     end
     CSV.write(save_name, df)
 end
 export save