refactor: use previously set violated constraint

NeuralNetworkVerification · Apr 4, 2018 · d0129c6 · d0129c6
1 parent 1e2bcae
commit d0129c6
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/engine/Engine.cpp b/src/engine/Engine.cpp
@@ -409,11 +409,7 @@ void Engine::performSimplexStep()
 
 void Engine::fixViolatedPlConstraintIfPossible()
 {
-    ASSERT( !_violatedPlConstraints.empty() );
-    PiecewiseLinearConstraint *violated = *_violatedPlConstraints.begin();
-    ASSERT( violated );
-
-    List<PiecewiseLinearConstraint::Fix> fixes = violated->getPossibleFixes();
+    List<PiecewiseLinearConstraint::Fix> fixes = _plConstraintToFix->getPossibleFixes();
 
     // First, see if we can fix without pivoting
     for ( const auto &fix : fixes )
@@ -655,7 +651,9 @@ bool Engine::allPlConstraintsHold()
 
 void Engine::selectViolatedPlConstraint()
 {
+    ASSERT( !_violatedPlConstraints.empty() );
     _plConstraintToFix = *_violatedPlConstraints.begin();
+    ASSERT( _plConstraintToFix );
 }
 
 void Engine::reportPlViolation()