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
Loading
Loading
Loading
Loading
+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 [
+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 = {