Add ray 1.0 support (#2084)

rlworkgroup · Oct 2, 2020 · c43eaf7 · c43eaf7
1 parent 70956d1
commit c43eaf7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/tf/trpo_swimmer_ray_sampler.py b/examples/tf/trpo_swimmer_ray_sampler.py
@@ -31,7 +31,7 @@ def trpo_swimmer_ray_sampler(ctxt=None, seed=1):
     """
     # Since this is an example, we are running ray in a reduced state.
     # One can comment this line out in order to run ray at full capacity
-    ray.init(memory=52428800,
+    ray.init(_memory=52428800,
              object_store_memory=78643200,
              ignore_reinit_error=True,
              log_to_driver=False,

diff --git a/examples/torch/trpo_pendulum_ray_sampler.py b/examples/torch/trpo_pendulum_ray_sampler.py
@@ -31,7 +31,7 @@ def trpo_pendulum_ray_sampler(ctxt=None, seed=1):
     """
     # Since this is an example, we are running ray in a reduced state.
     # One can comment this line out in order to run ray at full capacity
-    ray.init(memory=52428800,
+    ray.init(_memory=52428800,
              object_store_memory=78643200,
              ignore_reinit_error=True,
              log_to_driver=False,

diff --git a/tests/fixtures/sampler/ray_fixtures.py b/tests/fixtures/sampler/ray_fixtures.py
@@ -34,7 +34,7 @@ def ray_session_fixture():
 
     """
     if not ray.is_initialized():
-        ray.init(memory=52428800,
+        ray.init(_memory=52428800,
                  object_store_memory=78643200,
                  ignore_reinit_error=True,
                  log_to_driver=False,