"lhotse-speech#1203 Added handlings for negative end time"

ArthLeu · Jan 17, 2024 · c74739c · c74739c
1 parent 0089643
commit c74739c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lhotse/cut/set.py b/lhotse/cut/set.py
@@ -2874,15 +2874,21 @@ def compute_supervisions_frame_mask(
                     if ali.end < cut.duration
                     else num_frames
                 )
+                if et <= 0:
+                    continue
                 mask[st:et] = 1.0
+
         else:
             st = round(supervision.start / frame_shift) if supervision.start > 0 else 0
             et = (
                 round(supervision.end / frame_shift)
                 if supervision.end < cut.duration
                 else num_frames
             )
+            if et <= 0:
+                continue
             mask[st:et] = 1.0
+
     return mask