fixing logging in find_ioequations

SciML · Mar 10, 2024 · e1c8143 · e1c8143
1 parent 372a297
commit e1c8143
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/global_identifiability.jl b/src/global_identifiability.jl
@@ -106,8 +106,8 @@ The function returns a tuple containing the following:
     rational_interpolator = :VanDerHoevenLecerf,
 ) where {T}
     @info "Computing IO-equations"
-    ioeq_time =
-        @elapsed io_equations = find_ioequations(ode; var_change_policy = var_change_policy)
+    ioeq_time = @elapsed io_equations =
+        _find_ioequations(ode; var_change_policy = var_change_policy)
     @debug "Sizes: $(map(length, values(io_equations)))"
     @info "Computed IO-equations in $ioeq_time seconds"
     _runtime_logger[:ioeq_time] = ioeq_time

diff --git a/src/io_equation.jl b/src/io_equation.jl
@@ -346,6 +346,17 @@ Output:
     ode::ODE{P};
     var_change_policy = :default,
     loglevel = Logging.Info,
+) where {P <: MPolyRingElem{<:FieldElem}}
+    restart_logging(loglevel = loglevel)
+    reset_timings()
+    with_logger(_si_logger[]) do
+        return _find_ioequations(ode, var_change_policy = var_change_policy)
+    end
+end
+
+@timeit _to function _find_ioequations(
+    ode::ODE{P};
+    var_change_policy = :default,
 ) where {P <: MPolyRingElem{<:FieldElem}}
     # Setting the var_change policy
     if (var_change_policy == :yes) ||