From 0a6368616cdeca7525b3ea64b086b188c416ee60 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 30 Apr 2025 17:07:10 +0200
Subject: [PATCH] python312Packages.torchrl: 0.7.2 -> 0.8.0

Diff: https://github.com/pytorch/rl/compare/refs/tags/v0.7.2...refs/tags/v0.8.0

Changelog: https://github.com/pytorch/rl/releases/tag/v0.8.0
---
 .../python-modules/torchrl/default.nix        | 21 ++++++++++++++-----
 1 file changed, 16 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/torchrl/default.nix b/pkgs/development/python-modules/torchrl/default.nix
index 13e8a5870a22..f674b1135080 100644
--- a/pkgs/development/python-modules/torchrl/default.nix
+++ b/pkgs/development/python-modules/torchrl/default.nix
@@ -48,14 +48,14 @@
 
 buildPythonPackage rec {
   pname = "torchrl";
-  version = "0.7.2";
+  version = "0.8.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "rl";
     tag = "v${version}";
-    hash = "sha256-hcCZSASAp9jbOhbFLJndridYn76R99K+LxRxQl3uaxM=";
+    hash = "sha256-icT+QeA2FNhZjwD0ykui4aq5WswDv2i1QRh7dNlA4Cg=";
   };
 
   build-system = [
@@ -139,6 +139,20 @@ buildPythonPackage rec {
       # https://github.com/pytorch/rl/discussions/2483
       "test_resetting_strategies"
       "test_torchrl_to_gym"
+      "test_vecenvs_nan"
+
+      # gym.error.VersionNotFound: Environment version `v5` for environment `HalfCheetah` doesn't exist.
+      "test_collector_run"
+      "test_transform_inverse"
+
+      # OSError: Unable to synchronously create file (unable to truncate a file which is already open)
+      "test_multi_env"
+      "test_simple_env"
+
+      # ImportWarning: Ignoring non-library in plugin directory:
+      # /nix/store/cy8vwf1dacp3xfwnp9v6a1sz8bic8ylx-python3.12-mujoco-3.3.2/lib/python3.12/site-packages/mujoco/plugin/libmujoco.so.3.3.2
+      "test_auto_register"
+      "test_info_dict_reader"
 
       # mujoco.FatalError: an OpenGL platform library has not been loaded into this process, this most likely means that a valid OpenGL context has not been created before mjr_makeContext was called
       "test_vecenvs_env"
@@ -186,8 +200,5 @@ buildPythonPackage rec {
     changelog = "https://github.com/pytorch/rl/releases/tag/v${version}";
     license = lib.licenses.mit;
     maintainers = with lib.maintainers; [ GaetanLepage ];
-    # torhcrl is not compatible with our current version of gymnasium (>=1.0)
-    # https://github.com/pytorch/rl/pull/2473
-    broken = true;
   };
 }