Fixed SMG bug

randriu · Dec 6, 2024 · 45b863e · 45b863e
1 parent dc3ee84
commit 45b863e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/payntbind/src/synthesis/smg/modelchecker/helper/SparseSmgRpatlHelper.cpp b/payntbind/src/synthesis/smg/modelchecker/helper/SparseSmgRpatlHelper.cpp
@@ -160,6 +160,7 @@ namespace synthesis {
                         for (auto choice : transitionMatrix.getRowGroupIndices(state)) {
                             if (maximize ? epsilonGreaterOrEqual(constrainedChoiceValues[choice], result[state]) : epsilonGreaterOrEqual(result[state], constrainedChoiceValues[choice])) {
                                 optimalChoices[state] = stateRowIndex;
+                                optimalChoiceSet.set(state);
                                 break;
                             }
                             stateRowIndex++;
@@ -169,6 +170,7 @@ namespace synthesis {
                         for (auto choice : transitionMatrix.getRowGroupIndices(state)) {
                             if (maximize ? epsilonGreaterOrEqual(result[state], constrainedChoiceValues[choice]) : epsilonGreaterOrEqual(constrainedChoiceValues[choice], result[state])) {
                                 optimalChoices[state] = stateRowIndex;
+                                optimalChoiceSet.set(state);
                                 break;
                             }
                             stateRowIndex++;