python3Packages.numpyro: skip failing tests

This commit is contained in:
Gaetan Lepage
2025-12-29 23:46:59 +00:00
parent cbff2fd0c9
commit 21641cd006
@@ -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 <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.
"test_random_nnx_module_mcmc_sequence_param"
# AssertionError, assert GLOBAL["count"] == 4 (assert 5 == 4)
"test_mcmc_parallel_chain"