diff options
author | Gaetan Lepage <gaetan@glepage.com> | 2024-06-09 13:09:39 +0200 |
---|---|---|
committer | Gaetan Lepage <gaetan@glepage.com> | 2024-06-09 19:00:02 +0200 |
commit | c49c31fc90c90332c0564dfa59e6b325a877c419 (patch) | |
tree | f523020fc7c78cce1d501b3c7a143d7339446db6 /pkgs | |
parent | e26ad74eeeb5eb1b022f8b587c24f6d2d503a72d (diff) |
python311Packages.tensordict: disable flaky test on aarch64-linux
Diffstat (limited to 'pkgs')
-rw-r--r-- | pkgs/development/python-modules/tensordict/default.nix | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix index 30e98654d25c6..2fa2944a916d4 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"; }; |