Merge pull request #259131 from GaetanLepage/numpyro
python310Packages.numpyro: 0.13.0 -> 0.13.2
This commit is contained in:
commit
4b570045d7
1 changed files with 11 additions and 4 deletions
|
@ -9,19 +9,20 @@
|
||||||
, tqdm
|
, tqdm
|
||||||
, funsor
|
, funsor
|
||||||
, pytestCheckHook
|
, pytestCheckHook
|
||||||
, tensorflow-probability
|
# TODO: uncomment when tensorflow-probability gets fixed.
|
||||||
|
# , tensorflow-probability
|
||||||
}:
|
}:
|
||||||
|
|
||||||
buildPythonPackage rec {
|
buildPythonPackage rec {
|
||||||
pname = "numpyro";
|
pname = "numpyro";
|
||||||
version = "0.13.0";
|
version = "0.13.2";
|
||||||
format = "setuptools";
|
format = "setuptools";
|
||||||
|
|
||||||
disabled = pythonOlder "3.9";
|
disabled = pythonOlder "3.9";
|
||||||
|
|
||||||
src = fetchPypi {
|
src = fetchPypi {
|
||||||
inherit version pname;
|
inherit version pname;
|
||||||
hash = "sha256-n+5K6fZlatKkXGVxzKcVhmP5XNuJeeM+GcCJ1Kh/WMk=";
|
hash = "sha256-Um8LFVGAlMeOaN9uMwycHJzqEnTaxp8FYXIk+m2VTug=";
|
||||||
};
|
};
|
||||||
|
|
||||||
propagatedBuildInputs = [
|
propagatedBuildInputs = [
|
||||||
|
@ -35,7 +36,8 @@ buildPythonPackage rec {
|
||||||
nativeCheckInputs = [
|
nativeCheckInputs = [
|
||||||
funsor
|
funsor
|
||||||
pytestCheckHook
|
pytestCheckHook
|
||||||
tensorflow-probability
|
# TODO: uncomment when tensorflow-probability gets fixed.
|
||||||
|
# tensorflow-probability
|
||||||
];
|
];
|
||||||
|
|
||||||
pythonImportsCheck = [
|
pythonImportsCheck = [
|
||||||
|
@ -62,6 +64,11 @@ buildPythonPackage rec {
|
||||||
"test_model_transformation"
|
"test_model_transformation"
|
||||||
];
|
];
|
||||||
|
|
||||||
|
# TODO: remove when tensorflow-probability gets fixed.
|
||||||
|
disabledTestPaths = [
|
||||||
|
"test/test_distributions.py"
|
||||||
|
];
|
||||||
|
|
||||||
meta = with lib; {
|
meta = with lib; {
|
||||||
description = "Library for probabilistic programming with NumPy";
|
description = "Library for probabilistic programming with NumPy";
|
||||||
homepage = "https://num.pyro.ai/";
|
homepage = "https://num.pyro.ai/";
|
||||||
|
|
Loading…
Reference in a new issue