wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-19 11:27:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Core] Add option to override environment variables for tasks and actors (#11619)

Browse Source
This commit is contained in:
architkulkarni
2020-10-29 14:22:44 -05:00
committed by GitHub
parent e82ff08b0c
commit 4175569d96
16 changed files with 286 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/actor.py
+13 -4
View File
@@ -418,7 +418,8 @@ class ActorClass:
lifetime=None,
placement_group=None,
placement_group_bundle_index=-1,
placement_group_capture_child_tasks=None):
placement_group_capture_child_tasks=None,
override_environment_variables=None):
"""Configures and overrides the actor instantiation parameters.
The arguments are the same as those that can be passed
@@ -458,7 +459,9 @@ class ActorClass:
placement_group=placement_group,
placement_group_bundle_index=placement_group_bundle_index,
placement_group_capture_child_tasks=(
placement_group_capture_child_tasks))
placement_group_capture_child_tasks),
override_environment_variables=(
override_environment_variables))
return ActorOptionWrapper()
@@ -478,7 +481,8 @@ class ActorClass:
lifetime=None,
placement_group=None,
placement_group_bundle_index=-1,
placement_group_capture_child_tasks=None):
placement_group_capture_child_tasks=None,
override_environment_variables=None):
"""Create an actor.
This method allows more flexibility than the remote method because
@@ -515,6 +519,9 @@ class ActorClass:
placement_group_capture_child_tasks: Whether or not children tasks
of this actor should implicitly use the same placement group
as its parent. It is True by default.
override_environment_variables: Environment variables to override
and/or introduce for this actor. This is a dictionary mapping
variable names to their values.
Returns:
A handle to the newly created actor.
@@ -661,7 +668,9 @@ class ActorClass:
placement_group_bundle_index,
placement_group_capture_child_tasks,
# Store actor_method_cpu in actor handle's extension data.
extension_data=str(actor_method_cpu))
extension_data=str(actor_method_cpu),
override_environment_variables=override_environment_variables
or dict())
actor_handle = ActorHandle(
meta.language,