More fix for tune_parameters in FIRK

Author: ErikQQY

lib/BoundaryValueDiffEqFIRK/src/utils.jl

@@ -41,37 +41,37 @@ end
 @inline __K0_on_u0(
     prob,
     stage;
-    fit_parameters = false
-) = __K0_on_u0(prob, prob.u0, stage; fit_parameters = fit_parameters)
+    tune_parameters = false
+) = __K0_on_u0(prob, prob.u0, stage; tune_parameters = tune_parameters)
 @inline __K0_on_u0(
     prob,
     u0::AbstractArray,
     stage;
-    fit_parameters = false
-) = ifelse(fit_parameters, repeat(vcat(u0, __tunable_part(prob.p)), 1, stage), repeat(u0, 1, stage))
-@inline function __K0_on_u0(prob, u0::AbstractVector{<:AbstractArray}, stage; fit_parameters = false)
+    tune_parameters = false
+) = ifelse(tune_parameters, repeat(vcat(u0, __tunable_part(prob.p)), 1, stage), repeat(u0, 1, stage))
+@inline function __K0_on_u0(prob, u0::AbstractVector{<:AbstractArray}, stage; tune_parameters = false)
     u0_mat = hcat(u0...)
     avg_u0 = vec(sum(u0_mat, dims = 2)) / size(u0_mat, 2)
-    fit_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
+    tune_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
     return repeat(avg_u0, 1, stage)
 end
-@inline function __K0_on_u0(prob, u0::AbstractVectorOfArray, stage; fit_parameters = false)
+@inline function __K0_on_u0(prob, u0::AbstractVectorOfArray, stage; tune_parameters = false)
     u0_mat = hcat(u0.u...)
     avg_u0 = vec(sum(u0_mat, dims = 2)) / size(u0_mat, 2)
-    fit_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
+    tune_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
     return repeat(avg_u0, 1, stage)
 end
-@inline function __K0_on_u0(prob, u0::SciMLBase.ODESolution, stage; fit_parameters = false)
+@inline function __K0_on_u0(prob, u0::SciMLBase.ODESolution, stage; tune_parameters = false)
     u0_mat = hcat(u0.u...)
     avg_u0 = vec(sum(u0_mat, dims = 2)) / size(u0_mat, 2)
-    fit_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
+    tune_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
     return repeat(avg_u0, 1, stage)
 end
-@inline function __K0_on_u0(prob, u0::F, stage; fit_parameters = false) where {
+@inline function __K0_on_u0(prob, u0::F, stage; tune_parameters = false) where {
         F <:
         Function,
     }
     avg_u0 = u0(prob.p, first(prob.tspan))
-    fit_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
+    tune_parameters && return repeat(vcat(avg_u0, __tunable_part(prob.p)), 1, stage)
     return repeat(avg_u0, 1, stage)
 end