pass check_units through

Author: aml5600

src/systems/diffeqs/abstractodesystem.jl

@@ -778,6 +778,7 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
         build_initializeprob = true,
         initialization_eqs = [],
         fully_determined = false,
+        check_units = true,
         kwargs...)
     eqs = equations(sys)
     dvs = unknowns(sys)
@@ -816,7 +817,7 @@ function process_DEProblem(constructor, sys::AbstractODESystem, u0map, parammap;
         end
         initializeprob = ModelingToolkit.InitializationProblem(
             sys, t, u0map, parammap; guesses, warn_initialize_determined,
-            initialization_eqs, eval_expression, eval_module, fully_determined)
+            initialization_eqs, eval_expression, eval_module, fully_determined, check_units)
         initializeprobmap = getu(initializeprob, unknowns(sys))
 
         zerovars = Dict(setdiff(unknowns(sys), keys(defaults(sys))) .=> 0.0)
@@ -1426,18 +1427,19 @@ function InitializationProblem{iip, specialize}(sys::AbstractODESystem,
         warn_initialize_determined = true,
         initialization_eqs = [],
         fully_determined = false,
+        check_units = true,
         kwargs...) where {iip, specialize}
     if !iscomplete(sys)
         error("A completed system is required. Call `complete` or `structural_simplify` on the system before creating an `ODEProblem`")
     end
     if isempty(u0map) && get_initializesystem(sys) !== nothing
-        isys = get_initializesystem(sys; initialization_eqs)
+        isys = get_initializesystem(sys; initialization_eqs, check_units)
     elseif isempty(u0map) && get_initializesystem(sys) === nothing
         isys = structural_simplify(
-            generate_initializesystem(sys; initialization_eqs); fully_determined)
+            generate_initializesystem(sys; initialization_eqs, check_units); fully_determined)
     else
         isys = structural_simplify(
-            generate_initializesystem(sys; u0map, initialization_eqs); fully_determined)
+            generate_initializesystem(sys; u0map, initialization_eqs, check_units); fully_determined)
     end
 
     uninit = setdiff(unknowns(sys), [unknowns(isys); getfield.(observed(isys), :lhs)])

src/systems/nonlinear/initializesystem.jl

@@ -10,6 +10,7 @@ function generate_initializesystem(sys::ODESystem;
         default_dd_value = 0.0,
         algebraic_only = false,
         initialization_eqs = [],
+        check_units = true,
         kwargs...)
     sts, eqs = unknowns(sys), equations(sys)
     idxs_diff = isdiffeq.(eqs)
@@ -102,6 +103,7 @@ function generate_initializesystem(sys::ODESystem;
         pars;
         defaults = merge(ModelingToolkit.defaults(sys), todict(u0), dd_guess),
         parameter_dependencies = parameter_dependencies(sys),
+        checks = check_units,
         name,
         kwargs...)
 

test/dq_units.jl

@@ -207,5 +207,6 @@ end
     u0 = [pend.x => 1.0, pend.y => 0.0]
     p = [pend.g => 1.0, pend.L => 1.0]
     guess = [pend.λ => 0.0]
-    @test_broken prob = ODEProblem(pend, u0, (0.0, 1.0), p; guesses = guess)
+    @test prob = ODEProblem(
+        pend, u0, (0.0, 1.0), p; guesses = guess, check_units = false) isa Any
 end