Refactor frame number extraction logic

healthonrails · Nov 19, 2024 · 5e1850b · 5e1850b
1 parent bde669b
commit 5e1850b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/annolid/gui/app.py b/annolid/gui/app.py
@@ -1368,7 +1368,7 @@ def deleteAllFuturePredictions(self):
             if not prediction_file_path.endswith('.json'):
                 continue
 
-            frame_number = int(prediction_file_path.split(
+            frame_number = int(prediction_file.split(
                 '_')[-1].replace('.json', ''))
             is_future_frames = frame_number > self.frame_number
             is_manually_saved = os.path.exists(