fix ut

wenet-e2e · Jan 24, 2024 · 451f66a · 451f66a
1 parent 5211e4b
commit 451f66a
Showing 1 changed file with 1 addition and 1 deletion.
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