From f1f441590857cb6c2d256a33ee4eadf99f0697d4 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 22 Mar 2026 22:46:25 +0000
Subject: [PATCH] python3Packages.numpyro: skip failing tests

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

diff --git a/pkgs/development/python-modules/numpyro/default.nix b/pkgs/development/python-modules/numpyro/default.nix
index 45c0135317b7..cbdcd6f07edd 100644
--- a/pkgs/development/python-modules/numpyro/default.nix
+++ b/pkgs/development/python-modules/numpyro/default.nix
@@ -81,6 +81,12 @@ buildPythonPackage (finalAttrs: {
   ];
 
   disabledTests = [
+    # Failing with jax>=0.9.0
+    # TypeError: Error interpreting argument to closed_call as a JAX value
+    "test_provenance_call"
+    "test_provenance_closed_call"
+    "test_numpyrooptim_no_double_jit"
+
     # ValueError: Found unexpected Arrays on value of type <class 'list'> in static attribute 'layers'
     # of Pytree type '<class 'test_module.test_random_nnx_module_mcmc_sequence_params.<locals>.MLP'>'.
     # This is an error starting from Flax version 0.12.0.