Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.7 kB · 79 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980{ lib, buildPythonPackage, pythonOlder, fetchPypi, jax, jaxlib, multipledispatch, numpy, tqdm, funsor, pytestCheckHook# TODO: uncomment when tensorflow-probability gets fixed.# , tensorflow-probability}:
buildPythonPackage rec { pname = "numpyro"; version = "0.13.2"; format = "setuptools";
disabled = pythonOlder "3.9";
src = fetchPypi { inherit version pname; hash = "sha256-Um8LFVGAlMeOaN9uMwycHJzqEnTaxp8FYXIk+m2VTug="; };
propagatedBuildInputs = [ jax jaxlib multipledispatch numpy tqdm ];
nativeCheckInputs = [ funsor pytestCheckHook # TODO: uncomment when tensorflow-probability gets fixed. # tensorflow-probability ];
pythonImportsCheck = [ "numpyro" ];
disabledTests = [ # AssertionError due to tolerance issues "test_beta_binomial_log_prob" "test_collapse_beta" "test_cpu" "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" ];
# TODO: remove when tensorflow-probability gets fixed. disabledTestPaths = [ "test/test_distributions.py" ];
meta = with lib; { description = "Library for probabilistic programming with NumPy"; homepage = "https://num.pyro.ai/"; changelog = "https://github.com/pyro-ppl/numpyro/releases/tag/${version}"; license = licenses.asl20; maintainers = with maintainers; [ fab ]; };}