From f3776d75c853d56c31e035fdb81e10480535a203 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Mon, 14 Sep 2020 19:05:55 -0700 Subject: [PATCH] Restore plasma directory option (#10784) --- python/ray/worker.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/ray/worker.py b/python/ray/worker.py index 536f9d7a1..8061c5522 100644 --- a/python/ray/worker.py +++ b/python/ray/worker.py @@ -487,6 +487,7 @@ def init( # The following are unstable parameters and their use is discouraged. _enable_object_reconstruction=False, _redis_max_memory=None, + _plasma_directory=None, _node_ip_address=ray_constants.NODE_DEFAULT_IP, _driver_object_store_memory=None, _memory=None, @@ -573,6 +574,7 @@ def init( reconstructed. If False, then ray.ObjectLostError will be thrown. _redis_max_memory: Redis max memory. + _plasma_directory: Override the plasma mmap file directory. _node_ip_address (str): The IP address of the node that we are on. _driver_object_store_memory (int): Limit the amount of memory the driver can use in the object store for creating objects. @@ -668,7 +670,7 @@ def init( num_redis_shards=None, redis_max_clients=None, redis_password=_redis_password, - plasma_directory=None, + plasma_directory=_plasma_directory, huge_pages=None, include_dashboard=include_dashboard, dashboard_host=dashboard_host,