nixpkgs/pkgs/development/python-modules/pyro-ppl/default.nix
2023-01-16 11:14:09 +00:00

74 lines
1.1 KiB
Nix

{ lib
, buildPythonPackage
, fetchPypi
, graphviz
, jupyter
, matplotlib
, networkx
, opt-einsum
, pandas
, pillow
, pyro-api
, pythonOlder
, torch
, scikit-learn
, seaborn
, torchvision
, tqdm
, wget
}:
buildPythonPackage rec {
pname = "pyro-ppl";
version = "1.8.4";
format = "setuptools";
disabled = pythonOlder "3.7";
src = fetchPypi {
inherit version pname;
hash = "sha256-dm+tYeUt9IiF3pbUEhPaH46MG3ns8witUxifzRXBy0E=";
};
propagatedBuildInputs = [
pyro-api
torch
networkx
opt-einsum
tqdm
];
passthru.optional-dependencies = {
extras = [
graphviz
jupyter
# lap
matplotlib
pandas
pillow
scikit-learn
seaborn
torchvision
# visdom
wget
];
};
# pyro not shipping tests do simple smoke test instead
doCheck = false;
pythonImportsCheck = [
"pyro"
"pyro.distributions"
"pyro.infer"
"pyro.optim"
];
meta = with lib; {
description = "Library for probabilistic modeling and inference";
homepage = "http://pyro.ai";
license = licenses.asl20;
maintainers = with maintainers; [ teh georgewhewell ];
};
}