python312Packages.tensordict: enable now succeeding tests

2024-12-02 08:51:37 +01:00
parent 7746ea9889
commit 4e8f6593c0
1 changed files with 1 additions and 6 deletions
@@ -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