Add version of rllib for ppo script

LucasAlegre · Nov 27, 2024 · 973f169 · 973f169
1 parent 1045397
commit 973f169
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/experiments/ppo_4x4grid.py b/experiments/ppo_4x4grid.py
@@ -20,6 +20,14 @@
 
 
 if __name__ == "__main__":
+    # Use:
+    # ray[rllib]==2.7.0
+    # numpy == 1.23.4
+    # Pillow>=9.4.0
+    # ray[rllib]==2.7.0
+    # SuperSuit>=3.9.0
+    # torch>=1.13.1
+    # tensorflow-probability>=0.19.0
     ray.init()
 
     env_name = "4x4grid"

diff --git a/pyproject.toml b/pyproject.toml
@@ -52,7 +52,7 @@ nets = [
 [tool.black]
 safe = true
 line-length = 127
-target-version = ['py38', 'py39', 'py310', 'py311', 'py312']
+target-version = ['py38', 'py39', 'py310', 'py311']
 include = '\.pyi?$'
 
 [tool.isort]