Update test_reproducibility.py

pyannote · Jul 1, 2022 · 158e809 · 158e809
1 parent 7dc6296
commit 158e809
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/tasks/test_reproducibility.py b/tests/tasks/test_reproducibility.py
@@ -2,7 +2,7 @@
 import torch
 
 from pyannote.audio.models.segmentation.debug import SimpleSegmentationModel
-from pyannote.audio.tasks import MultilabelDetection, VoiceActivityDetection
+from pyannote.audio.tasks import MultiLabelSegmentation, VoiceActivityDetection
 from pyannote.database import FileFinder, get_protocol
 
 
@@ -32,7 +32,7 @@ def test_seeding_ensures_data_loaders():
     "Setting a global seed for the dataloaders ensures that we get data back in the same order"
     pl.seed_everything(1)
 
-    for task in [VoiceActivityDetection, MultilabelDetection]:
+    for task in [VoiceActivityDetection, MultiLabelSegmentation]:
         protocol, vad = setup_tasks(task)
         dl = create_dl(SimpleSegmentationModel, vad)
         last5a = get_next5(dl)
@@ -48,7 +48,7 @@ def test_seeding_ensures_data_loaders():
 def test_different_seeds():
     "Changing the global seed will change the order of the data that loads"
 
-    for task in [VoiceActivityDetection, MultilabelDetection]:
+    for task in [VoiceActivityDetection, MultiLabelSegmentation]:
         protocol, vad = setup_tasks(task)
         pl.seed_everything(4)
         dl = create_dl(SimpleSegmentationModel, vad)