fix: fix computation of same_chunk matrix

pyannote · Jan 28, 2022 · fd0c42c · fd0c42c
1 parent fc81b00
commit fd0c42c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pyannote/audio/pipelines/speaker_diarization.py b/pyannote/audio/pipelines/speaker_diarization.py
@@ -417,7 +417,10 @@ def oracle_cost_func(Y, y):
                         for i in permutation
                     ]
                 )
-                chunk_idx.extend([c] * len(permutation))
+                chunk_idx.extend([
+                        c if ((i is not None) and (active_reference[i])) else -1
+                        for i in permutation
+                ])
 
             oracle_clusters = np.array(oracle_clusters)
             oracle = 1.0 * squareform(pdist(oracle_clusters, metric="equal"))