fix: increase support to avoid rounding errors

pyannote · Jan 28, 2022 · fc81b00 · fc81b00
1 parent e4298a4
commit fc81b00
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pyannote/audio/pipelines/speaker_diarization.py b/pyannote/audio/pipelines/speaker_diarization.py
@@ -390,10 +390,9 @@ def oracle_cost_func(Y, y):
                 )
 
             chunks = segmentations.sliding_window
-            support = Segment(chunks[0].start, chunks[num_chunks - 1].end)
-
             reference = file["annotation"].discretize(
-                support=support, resolution=self._frames,
+                support=Segment(chunks[0].start, chunks[num_chunks - 1].end + chunks.step), 
+                resolution=self._frames,
             )
 
             oracle_clusters = []