Update pyannote/audio/pipelines/speech_separation.py

Co-authored-by: Hervé BREDIN <[email protected]>
pyannote · Oct 23, 2024 · 6de6952 · 6de6952
1 parent 3bf0cbd
commit 6de6952
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pyannote/audio/pipelines/speech_separation.py b/pyannote/audio/pipelines/speech_separation.py
@@ -701,11 +701,9 @@ def apply(
         # re-order sources so that they match
         # the order given by diarization.labels()
         inverse_mapping = {label: index for index, label in mapping.items()}
-        original_sliding_window = sources.sliding_window
-        data = sources.data[
+        source.data = sources.data[
             :, [inverse_mapping[label] for label in diarization.labels()]
         ]
-        sources = SlidingWindowFeature(data, original_sliding_window)
 
         if not return_embeddings:
             return diarization, sources