Skip to content

Commit 08be100

Browse files
committed
[UPD] Fix prune_cuts when only one cut is available
1 parent 1d1dc8a commit 08be100

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/SDDPoptimize.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -538,8 +538,12 @@ Return:
538538
function prune_cuts(model::SPModel, params::SDDPparameters, V::PolyhedralFunction)
539539
ncuts = V.numCuts
540540
# Find all active cuts:
541-
active_cuts = Bool[is_cut_active(model, i, V, params.solver) for i=1:ncuts]
542-
return PolyhedralFunction(V.betas[active_cuts], V.lambdas[active_cuts, :], sum(active_cuts))
541+
if ncuts > 1
542+
active_cuts = Bool[is_cut_active(model, i, V, params.solver) for i=1:ncuts]
543+
return PolyhedralFunction(V.betas[active_cuts], V.lambdas[active_cuts, :], sum(active_cuts))
544+
else
545+
return V
546+
end
543547
end
544548

545549

0 commit comments

Comments
 (0)