fix ut

wenet-e2e · Jan 24, 2024 · ca33aa5 · ca33aa5
1 parent 5211e4b
commit ca33aa5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/test/wenet/dataset/test_datapipes.py b/test/wenet/dataset/test_datapipes.py
@@ -34,4 +34,4 @@ def test_dynamic_batch_datapipe(data_list):
         padding)
 
     for d in dataset:
-        assert d['feats'].size(0) <= max_frames_in_batch
+        assert d['feats'].size(1) <= max_frames_in_batch
diff --git a/test/wenet/dataset/test_dataset.py b/test/wenet/dataset/test_dataset.py
@@ -19,7 +19,6 @@ def test_WenetRawDatasetSource(data_list):
     result = []
     for elem in dataset:
         result.append(elem)
-    print(result)
 
     assert len(result) == len(expected)
     for (i, elem) in enumerate(result):