diff --git a/python/ray/util/client/client_pickler.py b/python/ray/util/client/client_pickler.py index 99ea214fe00a..ca8624b6c5bc 100644 --- a/python/ray/util/client/client_pickler.py +++ b/python/ray/util/client/client_pickler.py @@ -150,11 +150,10 @@ def persistent_load(self, pid): def dumps_from_client(obj: Any, client_id: str, protocol=None) -> bytes: - with disable_client_hook(): - with io.BytesIO() as file: - cp = ClientPickler(client_id, file, protocol=protocol) - cp.dump(obj) - return file.getvalue() + with io.BytesIO() as file: + cp = ClientPickler(client_id, file, protocol=protocol) + cp.dump(obj) + return file.getvalue() def loads_from_server(data: bytes, diff --git a/python/requirements/tune/requirements_tune.txt b/python/requirements/tune/requirements_tune.txt index af25bf2aefa6..53990ab8e6d4 100644 --- a/python/requirements/tune/requirements_tune.txt +++ b/python/requirements/tune/requirements_tune.txt @@ -27,7 +27,7 @@ nevergrad==0.4.3.post3 optuna==2.6.0 pytest-remotedata==0.3.2 pytorch-lightning-bolts==0.2.5 -pytorch-lightning==1.0.3 +pytorch-lightning==1.3.8 scikit-learn==0.24.2 scikit-optimize==0.8.1 sigopt==7.4.0