mirror of
https://github.com/wassname/ray.git
synced 2026-07-03 12:19:50 +08:00
Introduce flag to use pickle for serialization (#5805)
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user