hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 18:11:42 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove recursive option from register_module (#232)

Browse source
This commit is contained in:
Robert Nishihara 2016-07-08 11:13:03 -07:00 • committed by Philipp Moritz
parent 7ff1580606
commit 82ab6d68f1
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lib/python/ray/worker.py
View file

@ -190,15 +190,19 @@ def task_info(worker=global_worker):
"""Tell the scheduler to return task information. Currently includes a list of all failed tasks since the start of the cluster."""
return ray.lib.task_info(worker.handle);
def register_module(module, recursive=False, worker=global_worker):
def register_module(module, worker=global_worker):
"""
This registers each remote function in the module with the scheduler, so tasks
with those functions can be scheduled on this worker.
:param module: The module of functions to register.
"""
logging.info("registering functions in module {}.".format(module.__name__))
for name in dir(module):
val = getattr(module, name)
if hasattr(val, "is_remote") and val.is_remote:
logging.info("registering {}.".format(val.func_name))
worker.register_function(val)
# elif recursive and isinstance(val, ModuleType):
# register_module(val, recursive, worker)
def connect(scheduler_address, objstore_address, worker_address, is_driver=False, worker=global_worker, mode=ray.WORKER_MODE):
if hasattr(worker, "handle"):