fixes to shell.py (#195)

This commit is contained in:
Robert Nishihara 2016-06-30 22:57:29 -07:00 committed by Philipp Moritz
parent 5821851318
commit 7611fbce4d

View file

@ -25,10 +25,10 @@ if __name__ == "__main__":
if args.attach:
assert args.worker_path is None, "when attaching, no new worker can be started"
assert args.num_workers is None, "when attaching, no new worker can be started"
worker.connect(args.scheduler_address, args.objstore_address, args.worker_address)
worker.connect(args.scheduler_address, args.objstore_address, args.worker_address, is_driver=True, mode=ray.SHELL_MODE)
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.worker_path is None else DEFAULT_WORKER_PATH,
driver_mode=ray.SHELL_MODE)
import IPython