diff --git a/python/ray/tests/test_dask_scheduler.py b/python/ray/tests/test_dask_scheduler.py index c818d65ca..318ffa83b 100644 --- a/python/ray/tests/test_dask_scheduler.py +++ b/python/ray/tests/test_dask_scheduler.py @@ -1,12 +1,15 @@ import dask import dask.array as da import pytest - +import sys +import unittest import ray -from ray.util.dask import ray_dask_get +@unittest.skipIf(sys.platform == "win32", "Failing on Windows.") def test_ray_dask_basic(ray_start_regular_shared): + from ray.util.dask import ray_dask_get + @ray.remote def stringify(x): return "The answer is {}".format(x) @@ -31,7 +34,9 @@ def test_ray_dask_basic(ray_start_regular_shared): assert ans == "The answer is 6", ans +@unittest.skipIf(sys.platform == "win32", "Failing on Windows.") def test_ray_dask_persist(ray_start_regular_shared): + from ray.util.dask import ray_dask_get arr = da.ones(5) + 2 result = arr.persist(scheduler=ray_dask_get) assert isinstance(next(iter(result.dask.values())), ray.ObjectRef) diff --git a/python/ray/tests/test_scheduling.py b/python/ray/tests/test_scheduling.py index 3a23e3f2c..d24f7eeb9 100644 --- a/python/ray/tests/test_scheduling.py +++ b/python/ray/tests/test_scheduling.py @@ -4,6 +4,7 @@ import logging import platform import sys import time +import unittest import numpy as np import pytest @@ -413,6 +414,7 @@ def test_locality_aware_leasing_borrowed_objects(ray_start_cluster): }).remote([f_obj])) == worker_node.unique_id +@unittest.skipIf(sys.platform == "win32", "Failing on Windows.") def test_lease_request_leak(shutdown_only): ray.init( num_cpus=1,