fix

ray-project · Dec 5, 2024 · f863a14 · f863a14
1 parent ecc8ef9
commit f863a14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ray/data/tests/test_webdataset.py b/python/ray/data/tests/test_webdataset.py
@@ -55,7 +55,7 @@ def test_webdataset_expand_json(ray_start_2_cpus, tmp_path):
 
     image = np.random.randint(0, 255, (100, 100, 3), dtype=np.uint8)
     gray = np.random.randint(0, 255, (100, 100), dtype=np.uint8)
-    dstruct = dict(a=[1], b=dict(c=2), d="hello")
+    dstruct = dict(a=[1,2], b=dict(c=2), d="hello")
     ttensor = torch.tensor([1, 2, 3]).numpy()
 
     sample = {
@@ -77,7 +77,7 @@ def test_webdataset_expand_json(ray_start_2_cpus, tmp_path):
         paths=[str(tmp_path)], override_num_blocks=1, expand_json=True
     )
     record = ds.take(1)
-    assert [1] == record[0]["a"][0]
+    assert [1,2] == record[0]["a"]
 
 
 def test_webdataset_suffixes(ray_start_2_cpus, tmp_path):