From 1e811148ed597bfe77793af5892f54a00622375f Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 25 Oct 2023 17:04:30 +0100
Subject: [PATCH] Fix dependency versions to align with GATE

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index d7e71a28..51ed84f2 100644
--- 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"