fix(reporter): error if video has an unstable end

williamfzc · Mar 2, 2020 · b1e222d · b1e222d
1 parent 5be2af0
commit b1e222d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/stagesepx/reporter.py b/stagesepx/reporter.py
@@ -221,12 +221,16 @@ def draw(
             else:
                 label = "unstable"
                 # add a frame
+                if cur_index + 1 < len(stage_range):
+                    new_each = [*each, stage_range[cur_index + 1][0]]
+                else:
+                    new_each = each
                 frame = np.hstack(
                     [
                         toolbox.compress_frame(
                             i.get_data(), compress_rate=compress_rate
                         )
-                        for i in [*each, stage_range[cur_index + 1][0]]
+                        for i in new_each
                     ]
                 )