delete reform at the end of optimize

atoptima · Mar 5, 2021 · a5cde53 · a5cde53
1 parent 4f1f039
commit a5cde53
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,6 +472,9 @@ 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,10 +50,6 @@ 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)
 
@@ -67,6 +63,7 @@ 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)