diff options
Diffstat (limited to 'pkgs/development/python-modules/tensordict/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tensordict/default.nix | 26 |
1 files changed, 16 insertions, 10 deletions
diff --git a/pkgs/development/python-modules/tensordict/default.nix b/pkgs/development/python-modules/tensordict/default.nix index a3b39969be6a1..bd35eff65707e 100644 --- a/pkgs/development/python-modules/tensordict/default.nix +++ b/pkgs/development/python-modules/tensordict/default.nix @@ -54,22 +54,28 @@ buildPythonPackage rec { pytestCheckHook ]; - # RuntimeError: internal error - disabledTests = lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [ - "test_add_scale_sequence" - "test_modules" - "test_setattr" - ]; + disabledTests = + # Hangs forever + [ "test_copy_onto" ] + ++ 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; { - description = "A pytorch dedicated tensor container"; + meta = { + description = "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"; }; |