Unverified Commit 41a0a66b authored by Martin Weinelt's avatar Martin Weinelt Committed by GitHub
Browse files

python312Packages.tensordict: disable flaky test_map_iter_interrupt_early on... 

python312Packages.tensordict: disable flaky test_map_iter_interrupt_early on all platforms (#361008)
parents 1e416e44 804b2e10

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

+4 −9
Original line number Diff line number Diff line
@@ -70,13 +70,11 @@ buildPythonPackage rec { 
      # +  where tensor(False) = <built-in method all of Tensor object at 0x7ffe49bf87d0>()

      "test_mp"



      # torch._dynamo.exc.BackendCompilerFailed

      # Requires a more recent version of triton

      # Re-enable when https://github.com/NixOS/nixpkgs/pull/328247 is merged

      # torch._dynamo.exc.InternalTorchDynamoError: RuntimeError: to_module requires TORCHDYNAMO_INLINE_INBUILT_NN_MODULES to be set.

      "test_functional"

      "test_linear"

      "test_seq"

      "test_seq_lmbda"



      # hangs forever on some CPUs

      "test_map_iter_interrupt_early"

    ]

    ++ lib.optionals (stdenv.hostPlatform.system == "aarch64-linux") [

      # RuntimeError: internal error
@@ -86,9 +84,6 @@ buildPythonPackage rec { 


      # _queue.Empty errors in multiprocessing tests

      "test_isend"



      # hangs forever

      "test_map_iter_interrupt_early"

    ];



  disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [

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

+4 −0
Original line number Diff line number Diff line
@@ -167,6 +167,10 @@ buildPythonPackage rec { 
    # assert torch.get_num_threads() == max(1, init_threads - 3)

    # AssertionError: assert 23 == 21

    "test_auto_num_threads"



    # Flaky (hangs indefinitely on some CPUs)

    "test_gae_multidim"

    "test_gae_param_as_tensor"

  ];



  meta = {