[data] Fix test failure caused by lack of ordring in default streamin…

…g executor (ray-project#34120) * Revert "[Datasets] Revert "Enable streaming executor by default (ray-project#32493)" (ray-project#33485)" This reverts commit 5c79954. * Fix test failure caused by lack of ordring in default streaming executor Signed-off-by: elliottower <[email protected]>
elliottower · Apr 22, 2023 · c8c2471 · c8c2471
1 parent 982a4c0
commit c8c2471
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/data/tests/test_dataset_consumption.py b/python/ray/data/tests/test_dataset_consumption.py
@@ -122,7 +122,7 @@ def test_dataset_lineage_serialization_unsupported(shutdown_only):
 
     serialized_ds = ds2.serialize_lineage()
     ds3 = Dataset.deserialize_lineage(serialized_ds)
-    assert ds3.take(30) == list(range(10)) + list(range(20))
+    assert set(ds3.take(30)) == set(list(range(10)) + list(range(20)))
 
     # Zips not supported.
     ds = ray.data.from_items(list(range(10)))