From c7f71748214a0f919d69014ef57d8dd2e8a18e75 Mon Sep 17 00:00:00 2001
From: ViktorM <victor.makoviychuk@gmail.com>
Date: Sat, 17 Oct 2020 10:10:12 -0700
Subject: [PATCH] Fixed Ray 1.0 API. Updated requirements.

---
 requirements.txt          | 6 +-----
 rl_games/tf14_runner.py   | 2 +-
 rl_games/torch_runner.py  | 2 +-
 setup.py                  | 2 +-
 test_a2c_continuous.ipynb | 4 ++--
 5 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index ab0496c9..8ac9d6fe 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,16 +4,12 @@ ray==1.0.0
 pyyaml
 setproctitle
 psutil
-torch>=1.5
+torch>=1.6
 tensorboard==1.14.0
 tensorboardX==1.6
-tensorflow-gpu==1.15.4
-tensorflow-probability==0.7.0
 opencv-python>=4.1.0.25
 gym-super-mario-bros==7.1.6
 pybullet>=2.5.0
 smac
 dm_control
 dm2gym
-
-
diff --git a/rl_games/tf14_runner.py b/rl_games/tf14_runner.py
index 0d592d0d..a3cc90ae 100644
--- a/rl_games/tf14_runner.py
+++ b/rl_games/tf14_runner.py
@@ -75,7 +75,7 @@ def get_prebuilt_config(self):
 
     def run_train(self):
         print('Started to train')
-        ray.init(redis_max_memory=1024*1024*1000, object_store_memory=1024*1024*1000)
+        ray.init(object_store_memory=1024*1024*1000)
         obs_space, action_space = env_configurations.get_obs_and_action_spaces_from_config(self.config)
         print('obs_space:', obs_space)
         print('action_space:', action_space)
diff --git a/rl_games/torch_runner.py b/rl_games/torch_runner.py
index a03a8cf1..5114161b 100644
--- a/rl_games/torch_runner.py
+++ b/rl_games/torch_runner.py
@@ -76,7 +76,7 @@ def get_prebuilt_config(self):
 
     def run_train(self):
         print('Started to train')
-        ray.init(redis_max_memory=1024*1024*1000, object_store_memory=1024*1024*1000)
+        ray.init(object_store_memory=1024*1024*1000)
         if self.exp_config:
             self.experiment =  experiment.Experiment(self.default_config, self.exp_config)
             exp_num = 0
diff --git a/setup.py b/setup.py
index f8bb7df5..2d2e6d4d 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
             # this setup is only for pytorch
             # 
             'gym>=0.17.2',
-            'torch>=1.5',
+            'torch>=1.6',
             'numpy>=1.16.0',
             'ray==1.0.0',
             'tensorboard>=1.14.0',
diff --git a/test_a2c_continuous.ipynb b/test_a2c_continuous.ipynb
index 5375621e..4f5353c7 100644
--- a/test_a2c_continuous.ipynb
+++ b/test_a2c_continuous.ipynb
@@ -130,7 +130,7 @@
    ],
    "source": [
     "import ray\n",
-    "ray.init(redis_max_memory=1024*1024*100, object_store_memory=1024*1024*100)"
+    "ray.init(object_store_memory=1024*1024*100)"
    ]
   },
   {
@@ -1048,4 +1048,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
\ No newline at end of file