Skip tests that use Python object Rows

Eventual-Inc · Jul 6, 2023 · 394bb25 · 394bb25
1 parent 5ecd004
commit 394bb25
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/ray/test_datasets.py b/tests/ray/test_datasets.py
@@ -61,6 +61,9 @@ def test_to_ray_dataset_all_arrow(n_partitions: int):
 
 
 @pytest.mark.skipif(get_context().runner_config.name != "ray", reason="Needs to run on Ray runner")
+@pytest.mark.skipif(
+    RAY_VERSION >= (2, 5, 0), reason="Ray Datasets versions >= 2.5.0 no longer support Python objects as rows"
+)
 @pytest.mark.parametrize("n_partitions", [1, 2])
 def test_to_ray_dataset_with_py(n_partitions: int):
     df = daft.from_pydict(DATA).repartition(n_partitions)