Hoist gamma check

CliMA · Sep 7, 2023 · b4ebb7e · b4ebb7e
1 parent 11e1f1a
commit b4ebb7e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/solvers/imex_ark.jl b/src/solvers/imex_ark.jl
@@ -55,12 +55,13 @@ function step_u!(integrator, cache::IMEXARKCache)
     s = length(b_exp)
 
     if !isnothing(T_imp!) && !isnothing(newtons_method)
+        γ isa Nothing && sdirk_error(name)
         NVTX.@range "update!" color = colorant"yellow" begin
             update!(
                 newtons_method,
                 newtons_method_cache,
                 NewTimeStep(t),
-                jacobian -> isnothing(γ) ? sdirk_error(name) : T_imp!.Wfact(jacobian, u, p, dt * γ, t),
+                jacobian -> T_imp!.Wfact(jacobian, u, p, dt * γ, t),
             )
         end
     end

diff --git a/src/solvers/imex_ssprk.jl b/src/solvers/imex_ssprk.jl
@@ -62,11 +62,12 @@ function step_u!(integrator, cache::IMEXSSPRKCache)
     s = length(b_imp)
 
     if !isnothing(T_imp!) && !isnothing(newtons_method)
+        γ isa Nothing && sdirk_error(name)
         update!(
             newtons_method,
             newtons_method_cache,
             NewTimeStep(t),
-            jacobian -> isnothing(γ) ? sdirk_error(name) : T_imp!.Wfact(jacobian, u, p, dt * γ, t),
+            jacobian -> T_imp!.Wfact(jacobian, u, p, dt * γ, t),
         )
     end