From 4e8f6593c0fe837b50facdf770807d036de3efd1 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 2 Dec 2024 08:51:37 +0100
Subject: [PATCH] python312Packages.tensordict: enable now succeeding tests

---
 pkgs/development/python-modules/tensordict/default.nix | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix
index 1ad591e13be6..9e7dd4e4ab56 100644
--- a/pkgs/development/python-modules/tensordict/default.nix
+++ b/pkgs/development/python-modules/tensordict/default.nix
@@ -70,13 +70,8 @@ buildPythonPackage rec {
       # +  where tensor(False) = <built-in method all of Tensor object at 0x7ffe49bf87d0>()
       "test_mp"
 
-      # torch._dynamo.exc.BackendCompilerFailed
-      # Requires a more recent version of triton
-      # Re-enable when https://github.com/NixOS/nixpkgs/pull/328247 is merged
+      # torch._dynamo.exc.InternalTorchDynamoError: RuntimeError: to_module requires TORCHDYNAMO_INLINE_INBUILT_NN_MODULES to be set.
       "test_functional"
-      "test_linear"
-      "test_seq"
-      "test_seq_lmbda"
     ]
     ++ lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [
       # RuntimeError: internal error