diff --git a/pkgs/development/python-modules/numpyro/default.nix b/pkgs/development/python-modules/numpyro/default.nix index 58d5604d0005..e4a9c508b8ee 100644 --- a/pkgs/development/python-modules/numpyro/default.nix +++ b/pkgs/development/python-modules/numpyro/default.nix @@ -75,9 +75,17 @@ buildPythonPackage rec { # Chains will be drawn sequentially. If you are running MCMC in CPU, consider using `numpyro.set_host_device_count(2)` at the beginning of your program. # You can double-check how many devices are available in your system using `jax.local_device_count()`. "-Wignore::UserWarning" + + # FutureWarning: In the future `np.object` will be defined as the corresponding NumPy scalar. + "-Wignore::FutureWarning" ]; disabledTests = [ + # ValueError: Found unexpected Arrays on value of type in static attribute 'layers' + # of Pytree type '.MLP'>'. + # This is an error starting from Flax version 0.12.0. + "test_random_nnx_module_mcmc_sequence_param" + # AssertionError, assert GLOBAL["count"] == 4 (assert 5 == 4) "test_mcmc_parallel_chain"