Bug fix for polynomial optimization for PERK integrator (#2004)

* fix bug for num_eval_stage + 1 * Update ext/TrixiConvexECOSExt.jl Co-authored-by: Daniel Doehring <[email protected]> --------- Co-authored-by: Daniel Doehring <[email protected]>
trixi-framework · Jul 8, 2024 · 0a74d75 · 0a74d75
1 parent 2d0aa80
commit 0a74d75
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ext/TrixiConvexECOSExt.jl b/ext/TrixiConvexECOSExt.jl
@@ -151,7 +151,14 @@ function Trixi.bisect_stability_polynomial(consistency_order, num_eig_vals,
         println("Concluded stability polynomial optimization \n")
     end
 
-    return evaluate(gamma), dt
+    gamma_opt = evaluate(gamma)
+
+    # Catch case S = 3 (only one opt. variable)
+    if isa(gamma_opt, Number)
+        gamma_opt = [gamma_opt]
+    end
+
+    return gamma_opt, dt
 end
 end # @muladd