Fix dependency versions to align with GATE

2023-10-25 17:04:30 +01:00
parent 2b8462d38d
commit 1e811148ed
1 changed files with 2 additions and 2 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,7 @@ classifiers = [
 ]

 dependencies = [
-    "gym==0.21.0",
+    "gymnasium==0.28.1",
    "jupyterlab==3.6.1",
    "kaleido==0.2.1",
    "matplotlib==3.7.1",
@@ -35,7 +35,7 @@ dependencies = [
    "polars==0.18.4",
    "prettytable==3.8.0",
    "PyYAML==6.0",
-    "stable-baselines3==1.6.2",
+    "stable-baselines3[extra]==2.1.0",
    "tensorflow==2.12.0",
    "typer[all]==0.9.0",
    "pydantic==2.1.1"