diff --git a/lightgbm_ray/tests/test_client.py b/lightgbm_ray/tests/test_client.py index 9cbdecd..f0d2f8b 100644 --- a/lightgbm_ray/tests/test_client.py +++ b/lightgbm_ray/tests/test_client.py @@ -19,6 +19,14 @@ def start_client_server_5_cpus(): yield client +@pytest.fixture +def start_client_server_5_cpus_modin(monkeypatch): + monkeypatch.setenv("__MODIN_AUTOIMPORT_PANDAS__", "1") + ray.init(num_cpus=5, runtime_env={"env_vars": {"__MODIN_AUTOIMPORT_PANDAS__": "1"}}) + with ray_start_client_server() as client: + yield client + + def test_simple_train(start_client_server_4_cpus): assert ray.util.client.ray.is_connected() from lightgbm_ray.examples.simple import main @@ -41,7 +49,7 @@ def test_simple_dask(start_client_server_5_cpus): main(cpus_per_actor=2, num_actors=2) -def test_simple_modin(start_client_server_5_cpus): +def test_simple_modin(start_client_server_5_cpus_modin): assert ray.util.client.ray.is_connected() from lightgbm_ray.examples.simple_modin import main