Revert "delete reform at the end of optimize"

This reverts commit a5cde53.
atoptima · Mar 5, 2021 · 37523d2 · 37523d2
1 parent a5cde53
commit 37523d2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/decomposition.jl b/src/decomposition.jl
@@ -472,9 +472,6 @@ function buildformulations!(
 end
 
 function reformulate!(prob::Problem, annotations::Annotations, env::Env)
-    if getcoefmatrix(prob.original_formulation).fillmode
-        closefillmode!(getcoefmatrix(prob.original_formulation))
-    end
     decomposition_tree = annotations.tree
     if decomposition_tree !== nothing
         root = BD.getroot(decomposition_tree)

diff --git a/src/optimize.jl b/src/optimize.jl
@@ -50,6 +50,10 @@ function optimize!(prob::MathProg.Problem, annotations::Annotations, params::Par
 
     env = Env(params)
 
+    if getcoefmatrix(prob.original_formulation).fillmode
+        closefillmode!(getcoefmatrix(prob.original_formulation))
+    end
+
     # Apply decomposition
     reformulate!(prob, annotations, env)
 
@@ -63,7 +67,6 @@ function optimize!(prob::MathProg.Problem, annotations::Annotations, params::Par
     end
 
     env.kpis.elapsed_optimization_time = elapsed_optim_time(env)
-    prob.re_formulation = nothing
 
     println(_to)
     TO.reset_timer!(_to)