Skip to content

Commit 1717e3c

Browse files
committed
Skip ray tests with newer dask version
Signed-off-by: Julio Faracco <[email protected]>
1 parent c548f6d commit 1717e3c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/pipeline/executors/test_ray.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,12 @@
55
import subprocess
66
import unittest
77

8+
import dask
89
import ray as ray_default
910
from mock import Mock, patch
11+
from packaging.version import Version
1012

1113
from dasf.pipeline.executors import RayPipelineExecutor
12-
from dasf.pipeline.executors.ray import setup_ray_protocol
1314
from dasf.utils.funcs import is_gpu_supported
1415

1516

@@ -30,6 +31,9 @@ def test_setup_ray_protocol_foo(self):
3031
self.assertTrue('Protocol foo is not supported.' in str(context.exception))
3132

3233

34+
@unittest.skipIf(Version(dask.__version__) >= Version('2024.12.1'),
35+
'There is a bug in ray with recent dask releases. '
36+
f'Current is {dask.__version__}')
3337
class TestRayExecutor(unittest.TestCase):
3438
def test_ray_executor_from_existing_nodes(self):
3539
proc = subprocess.run(["ray", "start", "--head", "--port=9191"],

0 commit comments

Comments
 (0)