From c49c31fc90c90332c0564dfa59e6b325a877c419 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 9 Jun 2024 13:09:39 +0200
Subject: [PATCH] python311Packages.tensordict: disable flaky test on
 aarch64-linux

---
 .../development/python-modules/tensordict/default.nix | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix
index 30e98654d25c..2fa2944a916d 100644
--- a/pkgs/development/python-modules/tensordict/default.nix
+++ b/pkgs/development/python-modules/tensordict/default.nix
@@ -57,22 +57,25 @@ buildPythonPackage rec {
   disabledTests =
     # Hangs forever
     [ "test_copy_onto" ]
-    # RuntimeError: internal error
     ++ 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"
     ];
 
   # ModuleNotFoundError: No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package
   disabledTestPaths = lib.optionals stdenv.isDarwin [ "test/test_distributed.py" ];
 
-  meta = with lib; {
+  meta = {
     description = "A pytorch dedicated tensor container";
     changelog = "https://github.com/pytorch/tensordict/releases/tag/v${version}";
     homepage = "https://github.com/pytorch/tensordict";
-    license = licenses.mit;
-    maintainers = with maintainers; [ GaetanLepage ];
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ GaetanLepage ];
     # No python 3.12 support yet: https://github.com/pytorch/rl/issues/2035
     broken = pythonAtLeast "3.12";
   };