From 00fd30401e1dd97bfeb14108c69deced2770e8a2 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 20 Jan 2026 13:01:28 +0000
Subject: [PATCH] python3Packages.torchdata: skip failing tests on python>=3.14

---
 .../python-modules/torchdata/default.nix      | 25 +++++++++++++------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/pkgs/development/python-modules/torchdata/default.nix b/pkgs/development/python-modules/torchdata/default.nix
index f74d9e159970..0466f6b68e00 100644
--- a/pkgs/development/python-modules/torchdata/default.nix
+++ b/pkgs/development/python-modules/torchdata/default.nix
@@ -19,9 +19,10 @@
   parameterized,
   pytest-xdist,
   pytestCheckHook,
+  pythonAtLeast,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "torchdata";
   version = "0.11.0";
   pyproject = true;
@@ -29,7 +30,7 @@ buildPythonPackage rec {
   src = fetchFromGitHub {
     owner = "meta-pytorch";
     repo = "data";
-    tag = "v${version}";
+    tag = "v${finalAttrs.version}";
     hash = "sha256-TSkZLL4WDSacuX4tl0+1bKSJCRI3LEhAyU3ztdlUvgk=";
   };
 
@@ -55,16 +56,24 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
-  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [
-    # RuntimeError: DataLoader timed out after 5 seconds
-    "test_ind_worker_queue"
-  ];
+  disabledTests =
+    lib.optionals (pythonAtLeast "3.14") [
+      # _pickle.PicklingError: Can't pickle local object <...worker_set_affinity at 0x7ffbcfa1ba00>
+      "test_set_affinity_in_worker_init"
+
+      # RuntimeError: DataLoader timed out after 5 seconds
+      "test_ind_worker_queue"
+    ]
+    ++ lib.optionals stdenv.hostPlatform.isDarwin [
+      # RuntimeError: DataLoader timed out after 5 seconds
+      "test_ind_worker_queue"
+    ];
 
   meta = {
     description = "Iterative enhancement to the PyTorch torch.utils.data.DataLoader and torch.utils.data.Dataset/IterableDataset";
     homepage = "https://github.com/meta-pytorch/data";
-    changelog = "https://github.com/meta-pytorch/data/releases/tag/${src.tag}";
+    changelog = "https://github.com/meta-pytorch/data/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.bsd3;
     maintainers = with lib.maintainers; [ GaetanLepage ];
   };
-}
+})