new ray 1.0 interface

wumpus · Nov 10, 2020 · 92bbb66 · 92bbb66
1 parent 9b3cba6
commit 92bbb66
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/paramsurvey/psray.py b/paramsurvey/psray.py
@@ -39,7 +39,7 @@ def init(system_kwargs, ncores=None, **kwargs):
     # should allow these to be kwargs
     address, password = read_ray_config()
     kwargs['address'] = address
-    kwargs['redis_password'] = password
+    kwargs['_redis_password'] = password  # added the leading _ in ray 1.0.0
 
     if os.environ.get('RAY_LOCAL_MODE', False):
         kwargs['local_mode'] = True

diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@
 ]
 
 extras_require = {
-    'ray': ['ray', 'pyarrow'],
+    'ray': ['ray>=1', 'pyarrow'],
     'mpi': ['mpi4py'],
     'test': test_requirements,  # setup no longer tests, so make them an extra that .travis.yml uses
 }

diff --git a/test/integration/test-ray.sh b/test/integration/test-ray.sh
@@ -15,7 +15,7 @@ echo $(hostname):$PORT $REDIS_PASSWORD > ~/.ray-test-72363726-details
 
 GIGABYTE=1000000000  # close enough
 
-ray start --head --redis-port=$PORT --redis-password=$REDIS_PASSWORD --memory $GIGABYTE --object-store-memory $GIGABYTE --redis-max-memory $GIGABYTE
+ray start --head --port=$PORT --redis-password=$REDIS_PASSWORD --memory $GIGABYTE --object-store-memory $GIGABYTE --redis-max-memory $GIGABYTE
 
 # in order to find both the uninstalled paramsurvey and our test program,
 # which ray needs to pickle, the PYTHONPATH must be manipulated. See