From c1c8be3c2ef57bfa38dfd1acece7125f31ab7205 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 10 Apr 2026 17:11:01 +0000
Subject: [PATCH] python3Packages.shimmy: 2.0.0 -> 2.0.1

Diff: https://github.com/Farama-Foundation/Shimmy/compare/v2.0.0...v2.0.1

Changelog: https://github.com/Farama-Foundation/Shimmy/releases/tag/v2.0.1
---
 .../python-modules/shimmy/default.nix         | 31 ++++---------------
 1 file changed, 6 insertions(+), 25 deletions(-)

diff --git a/pkgs/development/python-modules/shimmy/default.nix b/pkgs/development/python-modules/shimmy/default.nix
index 79df5decf66e..d807cb51fb98 100644
--- a/pkgs/development/python-modules/shimmy/default.nix
+++ b/pkgs/development/python-modules/shimmy/default.nix
@@ -2,7 +2,6 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
-  fetchpatch,
 
   # build-system
   setuptools,
@@ -21,32 +20,18 @@
   pytestCheckHook,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "shimmy";
-  version = "2.0.0";
+  version = "2.0.1";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "Farama-Foundation";
     repo = "Shimmy";
-    tag = "v${version}";
-    hash = "sha256-/wIXjOGb3UeMQdeifYagd7OcxbBcdGPS09mjvkFsWmk=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-vMVvD/UNzGs6igYskQ/L+yt0v5kdjOVal0sfaTkirsU=";
   };
 
-  patches = [
-    # Shimmy tries to register some environments from `dm-control` that require unpackaged `labmaze`.
-    # This prevents from importing `shimmy` itself by crashing with a `ModuleNotFoundError`.
-    # This patch imports those environments lazily.
-    #
-    # TODO: get rid of this patch at the next release as the issue has been fixed upstream:
-    # https://github.com/Farama-Foundation/Shimmy/pull/125
-    (fetchpatch {
-      name = "prevent-labmaze-import-crash";
-      url = "https://github.com/Farama-Foundation/Shimmy/commit/095d576f6aae15a09a1e426138629ce9f43a3c04.patch";
-      hash = "sha256-rr9l3tHunYFk0j7hfo9IaSRlogAtwXoXcQ0zuU/TL8c=";
-    })
-  ];
-
   build-system = [
     setuptools
   ];
@@ -76,10 +61,6 @@ buildPythonPackage rec {
     "tests/test_openspiel.py"
   ];
 
-  preCheck = ''
-    export HOME=$(mktemp -d)
-  '';
-
   disabledTests = [
     # Require network access
     "test_check_env[bsuite/mnist_noise-v0]"
@@ -103,10 +84,10 @@ buildPythonPackage rec {
   ];
 
   meta = {
-    changelog = "https://github.com/Farama-Foundation/Shimmy/releases/tag/v${version}";
+    changelog = "https://github.com/Farama-Foundation/Shimmy/releases/tag/${finalAttrs.src.tag}";
     description = "API conversion tool for popular external reinforcement learning environments";
     homepage = "https://github.com/Farama-Foundation/Shimmy";
     license = lib.licenses.mit;
     maintainers = with lib.maintainers; [ GaetanLepage ];
   };
-}
+})