Introduce flag to use pickle for serialization (#5805)

This commit is contained in:
Philipp Moritz
2019-10-18 22:29:36 -07:00
committed by GitHub
parent 29eee7f970
commit d23696de17
9 changed files with 85 additions and 22 deletions
+3
View File
@@ -74,6 +74,7 @@ class RayParams(object):
Java worker.
java_worker_options (str): The command options for Java worker.
load_code_from_local: Whether load code from local file or from GCS.
use_pickle: Whether data objects should be serialized with cloudpickle.
_internal_config (str): JSON configuration for overriding
RayConfig defaults. For testing purposes ONLY.
"""
@@ -113,6 +114,7 @@ class RayParams(object):
include_java=False,
java_worker_options=None,
load_code_from_local=False,
use_pickle=False,
_internal_config=None):
self.object_id_seed = object_id_seed
self.redis_address = redis_address
@@ -146,6 +148,7 @@ class RayParams(object):
self.include_java = include_java
self.java_worker_options = java_worker_options
self.load_code_from_local = load_code_from_local
self.use_pickle = use_pickle
self._internal_config = _internal_config
self._check_usage()