From f73539406fd5f27618e4d1e822b0a0f7aad6d11e Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 7 Jan 2025 00:10:09 +0100
Subject: [PATCH] python312Packages.numpyro: disable failing tests

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

diff --git a/pkgs/development/python-modules/numpyro/default.nix b/pkgs/development/python-modules/numpyro/default.nix
index 6e3e5ce6aa8e..f1f409d813c1 100644
--- a/pkgs/development/python-modules/numpyro/default.nix
+++ b/pkgs/development/python-modules/numpyro/default.nix
@@ -73,6 +73,10 @@ buildPythonPackage rec {
   ];
 
   disabledTests = [
+    # jax.errors.UnexpectedTracerError: Encountered an unexpected tracer
+    "test_haiku_state_dropout_smoke"
+    "test_flax_state_dropout_smoke"
+
     # AssertionError due to tolerance issues
     "test_beta_binomial_log_prob"
     "test_collapse_beta"