mirror of
https://github.com/wassname/ray.git
synced 2026-07-20 12:40:20 +08:00
[tune/core] serialization debugging utility (#12142)
Co-authored-by: SangBin Cho <rkooo567@gmail.com> Co-authored-by: Kai Fricke <kai@anyscale.com>
This commit is contained in:
co-authored by
SangBin Cho
Kai Fricke
parent
63b85df828
commit
a21523c709
@@ -1,7 +1,7 @@
|
||||
"""
|
||||
This file defines the common pytest fixtures used in current directory.
|
||||
"""
|
||||
|
||||
import os
|
||||
from contextlib import contextmanager
|
||||
import pytest
|
||||
import subprocess
|
||||
@@ -204,6 +204,13 @@ def call_ray_stop_only():
|
||||
subprocess.check_call(["ray", "stop"])
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def enable_pickle_debug():
|
||||
os.environ["RAY_PICKLE_VERBOSE_DEBUG"] = "1"
|
||||
yield
|
||||
del os.environ["RAY_PICKLE_VERBOSE_DEBUG"]
|
||||
|
||||
|
||||
@pytest.fixture()
|
||||
def two_node_cluster():
|
||||
system_config = {
|
||||
|
||||
Reference in New Issue
Block a user