From 220f8c6b5f84407a7ebd5e4b221b3b8d309b0043 Mon Sep 17 00:00:00 2001
From: Alexander Terekhov <alxtrkhv@gmail.com>
Date: Tue, 2 May 2023 09:16:12 +0300
Subject: [PATCH] Add missing freqai-rl dependencies

---
 setup.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 9c717df8b..512e112a8 100644
--- a/setup.py
+++ b/setup.py
@@ -21,9 +21,11 @@ freqai = [
 
 freqai_rl = [
     'torch',
+    'gymnasium',
     'stable-baselines3',
-    'gym==0.21',
-    'sb3-contrib'
+    'sb3-contrib',
+    'setuptools',
+    'tqdm'
 ]
 
 hdf5 = [