Skip client mode ut if ray > 2.4.0.

ray-project · Jan 17, 2024 · c706378 · c706378
1 parent cea87df
commit c706378
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/fed/tests/client_mode_tests/test_basic_client_mode.py b/fed/tests/client_mode_tests/test_basic_client_mode.py
@@ -19,8 +19,14 @@
 
 import fed
 import fed._private.compatible_utils as compatible_utils
+from fed._private.compatible_utils import _compare_version_strings
 from fed.tests.test_utils import ray_client_mode_setup # noqa
 
+pytest.mark.skipif(
+ _compare_version_strings(ray.__version__, '2.4.0', ),
+ reason='Skip client mode when ray > 2.4.0',
+)
+
 
 @fed.remote
 class MyModel: