Add a function to check nlls trait

SciML · Dec 21, 2023 · 12eab56 · 12eab56
1 parent 05ca0fc
commit 12eab56
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/problems/bvp_problems.jl b/src/problems/bvp_problems.jl
@@ -176,6 +176,16 @@ struct BVProblem{uType, tType, isinplace, nlls, P, F, PT, K} <:
     end
 end
 
+"""
+    isnonlinearleastsquares(prob::BVProblem)
+
+Returns `true` if the underlying problem is a nonlinear least squares problem.
+"""
+@inline function isnonlinearleastsquares(::BVProblem{uType,
+        tType, iip, nlls}) where {uType, tType, iip, nlls}
+    return nlls
+end
+
 struct FakeSolutionObject{U}
     u::U
 end