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

Merge pull request #179 from amplab/fix2

Browse source 
Fix imports for default_worker and set SHELL_MODE for shell
This commit is contained in:
Robert Nishihara 2016-06-27 17:32:08 -07:00 • committed by GitHub
commit 15a34fe9bf
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/default_worker.py
View file

@ -2,8 +2,8 @@ import sys
import argparse
import numpy as np
import ray.arrays.remote as ra
import ray.arrays.distributed as da
import ray.array.remote as ra
import ray.array.distributed as da
import ray
import ray.services as services

3
scripts/shell.py
View file

@ -28,7 +28,8 @@ if __name__ == "__main__":
worker.connect(args.scheduler_address, args.objstore_address, args.worker_address)
else:
services.start_ray_local(num_workers=args.num_workers if not args.num_workers is None else DEFAULT_NUM_WORKERS,
worker_path=args.worker_path if not args.num_workers is None else DEFAULT_WORKER_PATH)
worker_path=args.worker_path if not args.num_workers is None else DEFAULT_WORKER_PATH,
driver_mode=ray.SHELL_MODE)
import IPython
IPython.embed()