wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 01:16:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Return scheduler address in ray.init (#403)

Browse Source
This commit is contained in:
Richard Shin
2016-09-03 18:25:47 -07:00
committed by Robert Nishihara
parent a5df45a92e
commit efb61ca9c7
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/python/ray/worker.py
+7 -3
View File
@@ -609,9 +609,12 @@ def init(start_ray_local=False, num_workers=None, num_objstores=None, scheduler_
driver_mode (Optional[bool]): The mode in which to start the driver. This
should be one of SCRIPT_MODE, PYTHON_MODE, and SILENT_MODE.
raises:
Exception: An exception is raised if an inappropriate combination of
arguments is passed in.
Returns:
A string containing the address of the scheduler.
Raises:
Exception: An exception is raised if an inappropriate combination of
arguments is passed in.
"""
# Make GRPC only print error messages.
os.environ["GRPC_VERBOSITY"] = "ERROR"
@@ -643,6 +646,7 @@ def init(start_ray_local=False, num_workers=None, num_objstores=None, scheduler_
# to disconnect will happen in the call to cleanup() when the Python script
# exits.
connect(node_ip_address, scheduler_address, worker=global_worker, mode=driver_mode)
return scheduler_address
def cleanup(worker=global_worker):
"""Disconnect the driver, and terminate any processes started in init.