From 19b476004e07351933e646196294d04ad11ab3bb Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 8 Apr 2026 17:17:20 +0000
Subject: [PATCH] python3Packages.tensordict: skip failing tests on
 aarch64-linux

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

diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix
index 140aeafea321..81817baadfd3 100644
--- a/pkgs/development/python-modules/tensordict/default.nix
+++ b/pkgs/development/python-modules/tensordict/default.nix
@@ -89,6 +89,12 @@ buildPythonPackage (finalAttrs: {
     # AssertionError: assert 'a string!' == 'a metadata!'
     "test_save_load_memmap"
   ]
+  ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
+    # RuntimeError: Failed to initialize cpuinfo!
+    "test_cast_to"
+    "test_casts"
+    "test_td_params_cast"
+  ]
   ++ lib.optionals stdenv.hostPlatform.isDarwin [
     # Hangs due to the use of a pool
     "test_chunksize_num_chunks"