Fix the test.

kyutai-labs · Nov 6, 2024 · 3118696 · 3118696
1 parent 603250d
commit 3118696
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yomikomi-pyo3/test.py b/yomikomi-pyo3/test.py
@@ -136,7 +136,7 @@ def comparison_test(args):
 for v in yk.stream([1, 2, np.array([1, 2, 3, 4, 5])], field="foo"):
     print(v)
 for v in (
-    yk.stream(range(42), field="foo")
+    yk.stream([[i] for i in range(42)], field="foo")
     .sliding_window(7, field="foo", overlap_over_samples=True)
     .batch(2)
     .enumerate("idx")