Unverified Commit 344cecdc authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.torchdata: skip failing tests on python>=3.14 (#481944)

parents ce8c7e13 00fd3040

pkgs/development/python-modules/torchdata/default.nix

+17 −8
Original line number Diff line number Diff line
@@ -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,7 +56,15 @@ buildPythonPackage rec { 
    pytestCheckHook

  ];



  disabledTests = lib.optionals stdenv.hostPlatform.isDarwin [

  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"

    ];
@@ -63,8 +72,8 @@ buildPythonPackage rec { 
  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 ];

  };

}
 
})