[PEx] Minor correction

Author: aman-goel

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/runtime/scheduler/Schedule.java

@@ -126,12 +126,10 @@ public ScheduleChoice getScheduleChoiceAt(int choiceNum) {
     }
 
     /**
-     * Clear current choices at a choice depth
-     *
-     * @param idx Choice depth
+     * Clear current choices
      */
-    public void clearCurrent(int idx) {
-        choices.get(idx).clearCurrent();
+    public void clear() {
+        choices.clear();
     }
 
     /**

Src/PRuntimes/PExplicitRuntime/src/main/java/pexplicit/runtime/scheduler/explicit/ExplicitSearchScheduler.java

@@ -534,17 +534,18 @@ private void postIterationCleanup() {
             if (newStepNumber == 0) {
                 reset();
                 stepState.resetToZero();
+                schedule.clear();
             } else {
                 stepNumber = newStepNumber;
                 choiceNumber = scheduleChoice.getChoiceNumber();
                 stepState.setTo(scheduleChoice.getChoiceState());
                 assert (!PExplicitGlobal.getGlobalMachine(scheduleChoice.getCurrent()).getSendBuffer().isEmpty());
+                schedule.removeChoicesAfter(backtrackChoiceNumber);
             }
-            schedule.removeChoicesAfter(backtrackChoiceNumber);
             PExplicitLogger.logBacktrack(newStepNumber, cIdx, unit);
             return;
         }
-        schedule.removeChoicesAfter(0);
+        schedule.clear();
         isDoneIterating = true;
     }