Corrected the environment.yml file used for conda installation

edbeeching · Mar 24, 2022 · 1ab740e · 1ab740e
1 parent 6357c02
commit 1ab740e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -76,14 +76,14 @@ dependencies:
   - pip:
     - attrs==21.4.0
     - click==8.0.4
-    - cloudpickle==2.0.0
+    - cloudpickle==1.6.0
     - cycler==0.11.0
     - deprecated==1.2.13
     - dm-tree==0.1.6
     - filelock==3.6.0
     - fonttools==4.30.0
     - grpcio==1.43.0
-    - gym==0.21.0
+    - gym==0.19.0
     - imageio==2.16.1
     - importlib-resources==5.4.0
     - jsonschema==4.4.0
@@ -105,6 +105,7 @@ dependencies:
     - redis==4.1.4
     - scikit-image==0.19.2
     - scipy==1.8.0
+    - stable-baselines3==1.4.0
     - tabulate==0.8.9
     - tensorboardx==2.5
     - tifffile==2022.3.16