refactor: format

Author: AayushSabharwal

src/systems/abstractsystem.jl

@@ -352,8 +352,9 @@ function SymbolicIndexingInterface.is_variable(sys::AbstractSystem, sym)
         return sym in 1:length(variable_symbols(sys))
     end
     if has_index_cache(sys) && (ic = get_index_cache(sys)) !== nothing
-        return is_variable(ic, sym) || istree(sym) && operation(sym) === getindex &&
-            is_variable(ic, first(arguments(sym)))
+        return is_variable(ic, sym) ||
+               istree(sym) && operation(sym) === getindex &&
+               is_variable(ic, first(arguments(sym)))
     end
     return any(isequal(sym), variable_symbols(sys)) ||
            hasname(sym) && is_variable(sys, getname(sym))
@@ -379,8 +380,8 @@ function SymbolicIndexingInterface.variable_index(sys::AbstractSystem, sym)
         return if (idx = variable_index(ic, sym)) !== nothing
             idx
         elseif istree(sym) && operation(sym) === getindex &&
-            (idx = variable_index(ic, first(arguments(sym)))) !== nothing
-            idx[arguments(sym)[begin + 1:end]...]
+               (idx = variable_index(ic, first(arguments(sym)))) !== nothing
+            idx[arguments(sym)[(begin + 1):end]...]
         else
             nothing
         end
@@ -415,8 +416,9 @@ end
 function SymbolicIndexingInterface.is_parameter(sys::AbstractSystem, sym)
     sym = unwrap(sym)
     if has_index_cache(sys) && (ic = get_index_cache(sys)) !== nothing
-        return is_parameter(ic, sym) || istree(sym) && operation(sym) === getindex &&
-            is_parameter(ic, first(arguments(sym)))
+        return is_parameter(ic, sym) ||
+               istree(sym) && operation(sym) === getindex &&
+               is_parameter(ic, first(arguments(sym)))
     end
     if unwrap(sym) isa Int
         return unwrap(sym) in 1:length(parameter_symbols(sys))
@@ -441,8 +443,8 @@ function SymbolicIndexingInterface.parameter_index(sys::AbstractSystem, sym)
         return if (idx = parameter_index(ic, sym)) !== nothing
             idx
         elseif istree(sym) && operation(sym) === getindex &&
-            (idx = parameter_index(ic, first(arguments(sym)))) !== nothing
-            ParameterIndex(idx.portion, (idx.idx..., arguments(sym)[begin+1:end]...))
+               (idx = parameter_index(ic, first(arguments(sym)))) !== nothing
+            ParameterIndex(idx.portion, (idx.idx..., arguments(sym)[(begin + 1):end]...))
         else
             nothing
         end
@@ -1701,7 +1703,6 @@ function linearization_function(sys::AbstractSystem, inputs,
     u0, _p, _ = get_u0_p(sys, x0, p; use_union = false, tofloat = true)
     ps = parameters(sys)
     if has_index_cache(sys) && get_index_cache(sys) !== nothing
-            @show p full_parameters(sys)
         p = MTKParameters(sys, p)
     else
         p = _p

src/systems/index_cache.jl

@@ -160,10 +160,10 @@ end
 
 function SymbolicIndexingInterface.is_parameter(ic::IndexCache, sym)
     return check_index_map(ic.tunable_idx, sym) !== nothing ||
-        check_index_map(ic.discrete_idx, sym) !== nothing ||
-        check_index_map(ic.constant_idx, sym) !== nothing ||
-        check_index_map(ic.nonnumeric_idx, sym) !== nothing ||
-        check_index_map(ic.dependent_idx, sym) !== nothing
+           check_index_map(ic.discrete_idx, sym) !== nothing ||
+           check_index_map(ic.constant_idx, sym) !== nothing ||
+           check_index_map(ic.nonnumeric_idx, sym) !== nothing ||
+           check_index_map(ic.dependent_idx, sym) !== nothing
 end
 
 function SymbolicIndexingInterface.parameter_index(ic::IndexCache, sym)

src/systems/optimization/constraints_system.jl

@@ -182,7 +182,8 @@ function calculate_hessian(sys::ConstraintsSystem; sparse = false, simplify = fa
     return hess
 end
 
-function generate_hessian(sys::ConstraintsSystem, vs = unknowns(sys), ps = full_parameters(sys);
+function generate_hessian(
+        sys::ConstraintsSystem, vs = unknowns(sys), ps = full_parameters(sys);
         sparse = false, simplify = false, kwargs...)
     hess = calculate_hessian(sys, sparse = sparse, simplify = simplify)
     p = reorder_parameters(sys, ps)

src/systems/parameter_buffer.jl

@@ -38,7 +38,7 @@ function MTKParameters(sys::AbstractSystem, p; tofloat = false, use_union = fals
 
     for (sym, _) in p
         if istree(sym) && operation(sym) === getindex &&
-            first(arguments(sym)) in all_ps
+           first(arguments(sym)) in all_ps
             error("Scalarized parameter values ($sym) are not supported. Instead of `[p[1] => 1.0, p[2] => 2.0]` use `[p => [1.0, 2.0]]`")
         end
     end

test/initializationsystem.jl

@@ -429,4 +429,4 @@ sol = solve(prob, Tsit5())
 @test sol.u[1] == [0.0, 1.0]
 
 # This should warn, but logging tests can't be marked as broken
-@test_logs prob=ODEProblem(simpsys, [], tspan, guesses = [x => 2.0])
+@test_logs prob = ODEProblem(simpsys, [], tspan, guesses = [x => 2.0])