Update solve.jl

Added checks for requiresgradient, requireshessian, requiresconsjac, and requiresconshess in __check_opt_alg function.
SciML · Mar 19, 2024 · 718def1 · 718def1
1 parent 9fbff32
commit 718def1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/solve.jl b/src/solve.jl
@@ -116,6 +116,14 @@ function _check_opt_alg(prob::OptimizationProblem, alg; kwargs...)
         throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) requires constraints, pass them with the `cons` kwarg in `OptimizationFunction`."))
     !allowscallback(alg) && haskey(kwargs, :callback) &&
         throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) does not support callbacks, remove the `callback` keyword argument from the `solve` call."))
+    !requiresgradient(alg) && !(alg::Type{OptimizationFunction}) &&
+        throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) requires gradients, pass them in `OptimizationFunction`."))
+    !requireshessian(alg) && !(alg::Type{OptimizationFunction}) &&
+        throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) requires hessians, pass them in `OptimizationFunction`."))
+    !requiresconsjac(alg) && !(alg::Type{OptimizationFunction}) &&
+        throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) requires constraint jacobian, pass them with the `cons` kwarg in `OptimizationFunction`."))
+    !requiresconshess(alg) && !(alg::Type{OptimizationFunction}) &&
+        throw(IncompatibleOptimizerError("The algorithm $(typeof(alg)) requires constraint hessian, pass them with the `cons` kwarg in `OptimizationFunction`."))
     return
 end