From 8d70dd15dfc8ddadae47bb44ac6442e4164c7a2d Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Mon, 27 Jun 2016 17:23:01 -0700 Subject: [PATCH] Fix imports for default_worker and set SHELL_MODE for shell --- scripts/default_worker.py | 4 ++-- scripts/shell.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/scripts/default_worker.py b/scripts/default_worker.py index 0876d5479..2ee6b9542 100644 --- a/scripts/default_worker.py +++ b/scripts/default_worker.py @@ -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 diff --git a/scripts/shell.py b/scripts/shell.py index 39005ca40..0638ec88b 100644 --- a/scripts/shell.py +++ b/scripts/shell.py @@ -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()