add constructor for NonlinearAlias

SciML · Dec 10, 2024 · e1e45fd · e1e45fd
1 parent fb70e00
commit e1e45fd
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/src/problems/nonlinear_problems.jl b/src/problems/nonlinear_problems.jl
@@ -552,6 +552,17 @@ end
 
 
 struct NonlinearAliasSpecifier <: AbstractAliasSpecifier
-    alias_p
-    alias_f
+    alias_p::Union{Bool,Nothing}
+    alias_f::Union{Bool,Nothing}
+    alias_u0::Union{Bool,Nothing}
 end
+
+function NonlinearAliasSpecifier(;alias_p = nothing, alias_f = nothing, alias_u0 = nothing, alias = nothing)
+    if isnothing(alias)
+        NonlinearAliasSpecifier(alias_p, alias_f, alias_u0)
+    elseif alias
+        NonlinearAliasSpecifier(true, true, true)
+    elseif !alias
+        NonlinearAliasSpecifier(false, false, false)
+    end
+end