python312Packages.torchrl: skip flaky tests

2024-12-02 14:05:16 +01:00
parent e53e9be8c5
commit 804b2e10ec
1 changed files with 4 additions and 0 deletions
@@ -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 = {