wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-20 12:40:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

clean up imports (#230)

Browse Source
This commit is contained in:
Robert Nishihara
2016-07-08 12:46:47 -07:00
committed by Philipp Moritz
parent 191909dd93
commit 5dd411546d
16 changed files with 81 additions and 105 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_worker.py
+2 -4
View File
@@ -8,8 +8,6 @@ import ray.array.distributed as da
import ray.datasets.imagenet
import ray
import ray.services as services
import ray.worker as worker
parser = argparse.ArgumentParser(description='Parse addresses for the worker to connect to.')
parser.add_argument("--scheduler-address", default="127.0.0.1:10001", type=str, help="the scheduler's address")
@@ -18,7 +16,7 @@ parser.add_argument("--worker-address", default="127.0.0.1:40001", type=str, hel
if __name__ == "__main__":
args = parser.parse_args()
worker.connect(args.scheduler_address, args.objstore_address, args.worker_address)
ray.worker.connect(args.scheduler_address, args.objstore_address, args.worker_address)
ray.register_module(test_functions)
ray.register_module(ra)
@@ -29,4 +27,4 @@ if __name__ == "__main__":
ray.register_module(da.linalg)
ray.register_module(sys.modules[__name__])
worker.main_loop()
ray.worker.main_loop()