fix(classifier): stage range error

williamfzc · Mar 1, 2020 · 4511213 · 4511213
1 parent 22aef44
commit 4511213
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/stagesepx/classifier/base.py b/stagesepx/classifier/base.py
@@ -144,8 +144,9 @@ def get_stage_range(self) -> typing.List[typing.List[SingleClassifierResult]]:
         assert len(result) > 0, "video seems to only contain one stage"
 
         last = self.data[-1]
-        if result[-1][-1] != last:
-            result.append(self.data[cur.frame_id - 1 : last.frame_id - 1 + 1])
+        last_result = result[-1][-1]
+        if last_result != last:
+            result.append(self.data[last_result.frame_id - 1: last.frame_id - 1 + 1])
         return result
 
     def get_specific_stage_range(