diff --git a/pkgs/development/python-modules/stable-baselines3/default.nix b/pkgs/development/python-modules/stable-baselines3/default.nix
index c6a81f6be253..cb752b201af5 100644
--- a/pkgs/development/python-modules/stable-baselines3/default.nix
+++ b/pkgs/development/python-modules/stable-baselines3/default.nix
@@ -20,31 +20,20 @@
   rich,
   tqdm,
 }:
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "stable-baselines3";
-  version = "2.7.1";
+  version = "2.8.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "DLR-RM";
     repo = "stable-baselines3";
-    tag = "v${version}";
-    hash = "sha256-ucfdXyOYgevrKQ+RQbuoLjhGEvlzwH80yognMNbJlgQ=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-eMtkcPvTtdy0gqedCD8NxlC85rDEB9Dam5fIKujEWp4=";
   };
 
-  postPatch =
-    # Environment version v0 for `CliffWalking` is deprecated
-    ''
-      substituteInPlace "tests/test_vec_normalize.py" \
-        --replace-fail "CliffWalking-v0" "CliffWalking-v1"
-    '';
-
   build-system = [ setuptools ];
 
-  pythonRelaxDeps = [
-    "gymnasium"
-  ];
-
   dependencies = [
     cloudpickle
     gymnasium
@@ -83,8 +72,8 @@ buildPythonPackage rec {
   meta = {
     description = "PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms";
     homepage = "https://github.com/DLR-RM/stable-baselines3";
-    changelog = "https://github.com/DLR-RM/stable-baselines3/releases/tag/v${version}";
+    changelog = "https://github.com/DLR-RM/stable-baselines3/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.mit;
     maintainers = with lib.maintainers; [ derdennisop ];
   };
-}
+})