Skip to content

Commit f75a3a3

Browse files
committed
keep ray integration tests runnable after 2.55.1
1 parent 714c24f commit f75a3a3

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

tests/conftest.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3030,13 +3030,19 @@ def pyarrow_table_with_promoted_types(pyarrow_schema_with_promoted_types: "pa.Sc
30303030

30313031

30323032
@pytest.fixture(scope="session")
3033-
def ray_session() -> Generator[Any, None, None]:
3033+
def ray_session(tmp_path_factory: pytest.TempPathFactory) -> Generator[Any, None, None]:
30343034
"""Fixture to manage Ray initialization and shutdown for tests."""
3035+
os.environ["RAY_ENABLE_UV_RUN_RUNTIME_ENV"] = (
3036+
"0" # Disable Ray's uv-run runtime env propagation to keep test workers isolated.
3037+
)
3038+
30353039
import ray
30363040

3041+
working_dir = str(tmp_path_factory.mktemp("ray-working-dir"))
3042+
30373043
ray.init(
30383044
ignore_reinit_error=True,
3039-
runtime_env={"working_dir": None}, # Prevent Ray from serializing the working directory to workers
3045+
runtime_env={"working_dir": working_dir}, # Prevent Ray from serializing the working directory to workers
30403046
)
30413047
yield ray
30423048
ray.shutdown()

0 commit comments

Comments
 (0)