Commit 7452a2d1 authored by Fabian Affolter's avatar Fabian Affolter
Browse files

python310Packages.numpyro: add missing input for tests

parent afeb23c7

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

+7 −0
Original line number Diff line number Diff line
@@ -7,6 +7,7 @@ 
, numpy

, pytestCheckHook

, pythonOlder

, tensorflow-probability

, tqdm

}:
@@ -31,6 +32,7 @@ buildPythonPackage rec { 
  ];



  nativeCheckInputs = [

    tensorflow-probability

    pytestCheckHook

  ];
@@ -46,11 +48,16 @@ buildPythonPackage rec { 
    "test_gamma_poisson"

    "test_gof"

    "test_hpdi"

    "test_kl_dirichlet_dirichlet"

    "test_kl_univariate"

    "test_mean_var"

    # Tests want to download data

    "data_load"

    "test_jsb_chorales"

    # RuntimeWarning: overflow encountered in cast

    "test_zero_inflated_logits_probs_agree"

    # NameError: unbound axis name: _provenance

    "test_model_transformation"

  ];



  meta = with lib; {