Unverified Commit 4b570045 authored by OTABI Tomoya's avatar OTABI Tomoya Committed by GitHub
Browse files

Merge pull request #259131 from GaetanLepage/numpyro 

python310Packages.numpyro: 0.13.0 -> 0.13.2
parents 5b1c1674 9fca3a82

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

+11 −4
Original line number Diff line number Diff line
@@ -9,19 +9,20 @@ 
, tqdm

, funsor

, pytestCheckHook

, tensorflow-probability

# TODO: uncomment when tensorflow-probability gets fixed.

# , tensorflow-probability

}:



buildPythonPackage rec {

  pname = "numpyro";

  version = "0.13.0";

  version = "0.13.2";

  format = "setuptools";



  disabled = pythonOlder "3.9";



  src = fetchPypi {

    inherit version pname;

    hash = "sha256-n+5K6fZlatKkXGVxzKcVhmP5XNuJeeM+GcCJ1Kh/WMk=";

    hash = "sha256-Um8LFVGAlMeOaN9uMwycHJzqEnTaxp8FYXIk+m2VTug=";

  };



  propagatedBuildInputs = [
@@ -35,7 +36,8 @@ buildPythonPackage rec { 
  nativeCheckInputs = [

    funsor

    pytestCheckHook

    tensorflow-probability

    # TODO: uncomment when tensorflow-probability gets fixed.

    # tensorflow-probability

  ];



  pythonImportsCheck = [
@@ -62,6 +64,11 @@ buildPythonPackage rec { 
    "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/";