Formatting.

Author: lamorton

src/constants.jl

@@ -3,7 +3,7 @@ struct MTKConstantCtx end
 
 function isconstant(x)
     x = unwrap(x)
-    x isa Symbolic &&  getmetadata(x, MTKConstantCtx, false)
+    x isa Symbolic && getmetadata(x, MTKConstantCtx, false)
 end
 isconstant(x::Num) = isconstant(unwrap(x))
 """
@@ -17,7 +17,8 @@ function toconstant(s)
     elseif s isa AbstractArray
         map(toconstant, s)
     else
-        hasmetadata(s, Symbolics.VariableDefaultValue) || throw(ArgumentError("Constant `$(s)` must be assigned a default value."))
+        hasmetadata(s, Symbolics.VariableDefaultValue) ||
+            throw(ArgumentError("Constant `$(s)` must be assigned a default value."))
         setmetadata(s, MTKConstantCtx, true)
     end
 end

src/systems/diffeqs/abstractodesystem.jl

@@ -113,7 +113,8 @@ function generate_dae_jacobian(sys::AbstractODESystem, dvs = states(sys),
     @variables ˍ₋gamma
     jac = ˍ₋gamma * jac_du + jac_u
     pre = get_preprocess_constants(jac)
-    return build_function(jac, derivatives, dvs, ps, ˍ₋gamma, get_iv(sys); postprocess_fbody = pre, kwargs...)
+    return build_function(jac, derivatives, dvs, ps, ˍ₋gamma, get_iv(sys);
+                          postprocess_fbody = pre, kwargs...)
 end
 
 function generate_function(sys::AbstractODESystem, dvs = states(sys), ps = parameters(sys);

src/systems/diffeqs/odesystem.jl

@@ -238,7 +238,6 @@ function ODESystem(eqs, iv = nothing; kwargs...)
                      collect(Iterators.flatten((diffvars, algevars))), ps; kwargs...)
 end
 
-
 # NOTE: equality does not check cached Jacobian
 function Base.:(==)(sys1::ODESystem, sys2::ODESystem)
     sys1 === sys2 && return true

src/systems/systemstructure.jl

@@ -266,7 +266,8 @@ function TearingState(sys; quick_cancel = false, check = true)
         for var in vars
             _var, _ = var_from_nested_derivative(var)
             any(isequal(_var), ivs) && continue
-            if isparameter(_var) || (istree(_var) && isparameter(operation(_var)) || isconstant(_var))
+            if isparameter(_var) ||
+                (istree(_var) && isparameter(operation(_var)) || isconstant(_var))
                 continue
             end
             varidx = addvar!(var)

src/utils.jl

@@ -513,7 +513,7 @@ function collect_constants(eqs::Vector{Equation}) #For get_substitutions_and_sol
     return constants
 end
 
-function collect_constants(eqs::AbstractArray{T}) where T # For generate_tgrad / generate_jacobian / generate_difference_cb
+function collect_constants(eqs::AbstractArray{T}) where {T} # For generate_tgrad / generate_jacobian / generate_difference_cb
     constants = T[]
     for eq in eqs
         collect_constants!(constants, unwrap(eq))
@@ -542,7 +542,7 @@ end
 function get_preprocess_constants(eqs)
     cs = collect_constants(eqs)
     pre = ex -> Let(Assignment[Assignment(x, getdefault(x)) for x in cs],
-                           ex, false)
+                    ex, false)
     return pre
 end
 
@@ -586,7 +586,7 @@ end
 function get_substitutions_and_solved_states(sys; no_postprocess = false)
     #Inject substitutions for constants => values
     cs = collect_constants([sys.eqs; sys.observed]) #ctrls? what else?
-    if !empty_substitutions(sys) 
+    if !empty_substitutions(sys)
         cs = [cs; collect_constants(sys.substitutions.subs)]
     end
     # Swap constants for their values
@@ -598,7 +598,7 @@ function get_substitutions_and_solved_states(sys; no_postprocess = false)
     else # Have to do some work
         if !empty_substitutions(sys)
             @unpack subs = get_substitutions(sys)
-        else 
+        else
             subs = []
         end
         subs = [cmap; subs] # The constants need to go first

test/constants.jl

@@ -21,5 +21,4 @@ simp = structural_simplify(sys);
 @test isequal(equations(simp)[1], eqs[1])
 prob = ODEProblem(simp, [0, ], [0.0, 1.0], [])
 sol = solve(prob, Tsit5())
-@test sol[w][1] == 1 
-
+@test sol[w][1] == 1