diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix
index e9754a0a7817..29c92e118b36 100644
--- a/pkgs/development/python-modules/tensordict/default.nix
+++ b/pkgs/development/python-modules/tensordict/default.nix
@@ -13,7 +13,7 @@
   orjson,
   packaging,
 
-  # checks
+  # tests
   h5py,
   pytestCheckHook,
 
@@ -22,14 +22,14 @@
 
 buildPythonPackage rec {
   pname = "tensordict";
-  version = "0.7.1";
+  version = "0.7.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "tensordict";
     tag = "v${version}";
-    hash = "sha256-tUm1uV9k/IDlR3y/e1aIlU1bjDeh2+zdJdxu8Z9x3es=";
+    hash = "sha256-ZDfRvfyBashU4kIoo8JX/EoCv4tpDOyggOlpdVCudT8=";
   };
 
   build-system = [
@@ -57,35 +57,14 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
-  disabledTests =
-    [
-      # FileNotFoundError: [Errno 2] No such file or directory: '/build/source/tensordict/tensorclass.pyi
-      "test_tensorclass_instance_methods"
-      "test_tensorclass_stub_methods"
+  disabledTests = [
+    # FileNotFoundError: [Errno 2] No such file or directory: '/build/source/tensordict/tensorclass.pyi
+    "test_tensorclass_instance_methods"
+    "test_tensorclass_stub_methods"
 
-      # Hangs forever
-      "test_copy_onto"
-
-      # EOFError (MPI related)
-      # AssertionError: assert tensor(False)
-      # +  where tensor(False) = <built-in method all of Tensor object at 0x7ffe49bf87d0>()
-      "test_mp"
-
-      # torch._dynamo.exc.InternalTorchDynamoError: RuntimeError: to_module requires TORCHDYNAMO_INLINE_INBUILT_NN_MODULES to be set.
-      "test_functional"
-
-      # hangs forever on some CPUs
-      "test_map_iter_interrupt_early"
-    ]
-    ++ lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [
-      # RuntimeError: internal error
-      "test_add_scale_sequence"
-      "test_modules"
-      "test_setattr"
-
-      # _queue.Empty errors in multiprocessing tests
-      "test_isend"
-    ];
+    # hangs forever on some CPUs
+    "test_map_iter_interrupt_early"
+  ];
 
   disabledTestPaths =
     [