Fix imports for default_worker and set SHELL_MODE for shell

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

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()