Commit f45e8a3d authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent c42f42b6
Loading
Loading
Loading
Loading
+9 −6
Original line number Diff line number Diff line
@@ -28,16 +28,16 @@
  tensorflow-probability,
}:

buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
  pname = "numpyro";
  version = "0.19.0";
  version = "0.20.0";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "pyro-ppl";
    repo = "numpyro";
    tag = version;
    hash = "sha256-3kzaINsz1Mjk97ERQsQIYIBz7CVmXtVDn0edJFMHQWs=";
    tag = finalAttrs.version;
    hash = "sha256-lMga+mPQEh6RCeqXKa2KELR6RcksKJ/K32h7X7a1IcQ=";
  };

  build-system = [ setuptools ];
@@ -102,6 +102,9 @@ buildPythonPackage rec {

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

    # Failed: DID NOT RAISE <class 'UserWarning'>
    "test_interval_censored_validate_sample"
  ]
  ++ lib.optionals stdenv.hostPlatform.isDarwin [
    # AssertionError: Not equal to tolerance rtol=0.06, atol=0
@@ -116,8 +119,8 @@ buildPythonPackage rec {
  meta = {
    description = "Library for probabilistic programming with NumPy";
    homepage = "https://num.pyro.ai/";
    changelog = "https://github.com/pyro-ppl/numpyro/releases/tag/${version}";
    changelog = "https://github.com/pyro-ppl/numpyro/releases/tag/${finalAttrs.src.tag}";
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ fab ];
  };
}
})