[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:
Richard Liaw
2020-12-02 00:52:17 -08:00
committed by GitHub
co-authored by SangBin Cho Kai Fricke
parent 63b85df828
commit a21523c709
12 changed files with 394 additions and 24 deletions
+8 -1
View File
@@ -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 = {