diff --git a/doc/requirements-doc.txt b/doc/requirements-doc.txt index 818e570e2..b8ee6b358 100644 --- a/doc/requirements-doc.txt +++ b/doc/requirements-doc.txt @@ -23,4 +23,3 @@ flask uvicorn pygments werkzeug -cloudpickle diff --git a/python/ray/experimental/serve/kv_store_service.py b/python/ray/experimental/serve/kv_store_service.py index 423c572f7..05f8cb138 100644 --- a/python/ray/experimental/serve/kv_store_service.py +++ b/python/ray/experimental/serve/kv_store_service.py @@ -2,7 +2,7 @@ import json import sqlite3 from abc import ABC -import cloudpickle as pickle +from ray import cloudpickle as pickle import ray.experimental.internal_kv as ray_kv from ray.experimental.serve.utils import logger