Update cost_functions.jl

SciML · Jul 21, 2022 · ff5e092 · ff5e092
1 parent d999522
commit ff5e092
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cost_functions.jl b/src/cost_functions.jl
@@ -64,7 +64,7 @@ function (f::L2Loss)(sol::DiffEqBase.AbstractNoTimeSolution)
     sumsq
 end
 
-function (f::L2Loss)(sol::DiffEqBase.AbstractDESolution)
+function (f::L2Loss)(sol::DiffEqBase.AbstractSciMLSolution)
     data = f.data
     weight = f.data_weight
     diff_weight = f.differ_weight
@@ -168,7 +168,7 @@ function LogLikeLoss(t, data_distributions, diff_distributions)
     LogLikeLoss(t, matrixize(data_distributions), matrixize(diff_distributions), 1)
 end
 
-function (f::LogLikeLoss)(sol::SciMLBase.AbstractDESolution)
+function (f::LogLikeLoss)(sol::SciMLBase.AbstractSciMLSolution)
     distributions = f.data_distributions
     if sol isa DiffEqBase.AbstractEnsembleSolution
         failure = any(s.retcode !== :Success && s.retcode !== :Terminated for s in sol)