Commit 87553253 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.numpyro: skip newly failing tests 

https://github.com/pyro-ppl/numpyro/issues/2055
parent 6e4d454b

pkgs/development/python-modules/numpyro/default.nix

+5 −0
Original line number Diff line number Diff line
@@ -92,6 +92,11 @@ buildPythonPackage rec { 


    # ValueError: compiling computation that requires 2 logical devices, but only 1 XLA devices are available (num_replicas=2)

    "test_chain"



    # Failing since flax==0.11.0

    # KeyError: "No RngStream named 'dropout' found in Rngs."

    # https://github.com/pyro-ppl/numpyro/issues/2055

    "test_nnx_state_dropout_smoke"

  ]

  ++ lib.optionals stdenv.hostPlatform.isDarwin [

    # AssertionError: Not equal to tolerance rtol=0.06, atol=0