Commit 21641cd0 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.numpyro: skip failing tests

parent cbff2fd0
Loading
Loading
Loading
Loading
+8 −0
Original line number Diff line number Diff line
@@ -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"