From 804b2e10ec90e9b3c0abe9390b35bffc809ae441 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 2 Dec 2024 14:05:16 +0100
Subject: [PATCH] python312Packages.torchrl: skip flaky tests

---
 pkgs/development/python-modules/torchrl/default.nix | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkgs/development/python-modules/torchrl/default.nix b/pkgs/development/python-modules/torchrl/default.nix
index 990e4df76be4..a1555b29357e 100644
--- a/pkgs/development/python-modules/torchrl/default.nix
+++ b/pkgs/development/python-modules/torchrl/default.nix
@@ -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 = {