mirror of
https://github.com/SebastianWendel/nixpkgs.git
synced 2024-09-23 05:35:50 +02:00
python310Packages.distrax: init at 0.1.2
This commit is contained in:
parent
f22a2437f3
commit
b306581d10
57
pkgs/development/python-modules/distrax/default.nix
Normal file
57
pkgs/development/python-modules/distrax/default.nix
Normal file
|
@ -0,0 +1,57 @@
|
|||
{ lib
|
||||
, fetchPypi
|
||||
, buildPythonPackage
|
||||
, numpy
|
||||
, tensorflow-probability
|
||||
, chex
|
||||
, dm-haiku
|
||||
, pytestCheckHook
|
||||
, jaxlib }:
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "distrax";
|
||||
version = "0.1.2";
|
||||
|
||||
src = fetchPypi {
|
||||
inherit pname version;
|
||||
sha256 = "sha256-b/+rxjdowNMuhUBhRCuN45z/iUbj1hN1qCSQqqAtZIw=";
|
||||
};
|
||||
|
||||
buildInputs = [
|
||||
chex
|
||||
jaxlib
|
||||
numpy
|
||||
tensorflow-probability
|
||||
];
|
||||
|
||||
checkInputs = [
|
||||
dm-haiku
|
||||
pytestCheckHook
|
||||
];
|
||||
|
||||
pythonImportsCheck = [
|
||||
"distrax"
|
||||
];
|
||||
|
||||
disabledTestPaths = [
|
||||
# TypeErrors
|
||||
"distrax/_src/bijectors/tfp_compatible_bijector_test.py"
|
||||
"distrax/_src/distributions/distribution_from_tfp_test.py"
|
||||
"distrax/_src/distributions/laplace_test.py"
|
||||
"distrax/_src/distributions/multinomial_test.py"
|
||||
"distrax/_src/distributions/mvn_diag_plus_low_rank_test.py"
|
||||
"distrax/_src/distributions/mvn_kl_test.py"
|
||||
"distrax/_src/distributions/straight_through_test.py"
|
||||
"distrax/_src/distributions/tfp_compatible_distribution_test.py"
|
||||
"distrax/_src/distributions/transformed_test.py"
|
||||
"distrax/_src/distributions/uniform_test.py"
|
||||
"distrax/_src/utils/transformations_test.py"
|
||||
];
|
||||
|
||||
meta = with lib; {
|
||||
description = "Probability distributions in JAX";
|
||||
homepage = "https://github.com/deepmind/distrax";
|
||||
license = licenses.asl20;
|
||||
maintainers = with maintainers; [ onny ];
|
||||
};
|
||||
}
|
|
@ -2322,6 +2322,8 @@ in {
|
|||
|
||||
distorm3 = callPackage ../development/python-modules/distorm3 { };
|
||||
|
||||
distrax = callPackage ../development/python-modules/distrax { };
|
||||
|
||||
distributed = callPackage ../development/python-modules/distributed { };
|
||||
|
||||
distro = callPackage ../development/python-modules/distro { };
|
||||
|
|
Loading…
Reference in a new issue