Add missing freqai-rl dependencies

2026-01-20 14:00:38 +00:00 · 2023-05-02 09:16:12 +03:00
parent e3f983729f
commit 220f8c6b5f
1 changed files with 4 additions and 2 deletions
--- 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 = [