tjh.dev
1{ buildPythonPackage, fetchPypi, lib, pytorch, contextlib2
2, graphviz, networkx, six, opt-einsum, tqdm }:
3buildPythonPackage rec {
4 version = "1.1.0";
5 pname = "pyro-ppl";
6
7 src = fetchPypi {
8 inherit version pname;
9 sha256 = "5ca2fd19276fcfcf52babb48d22892a41d6238d7a6c65e63f704b070a3816479";
10 };
11
12 propagatedBuildInputs = [
13 pytorch
14 contextlib2
15 # TODO(tom): graphviz pulls in a lot of dependencies - make
16 # optional when some time to figure out how.
17 graphviz
18 networkx
19 six
20 opt-einsum
21 tqdm
22 ];
23
24 # pyro not shipping tests do simple smoke test instead
25 checkPhase = ''
26 python -c "import pyro"
27 python -c "import pyro.distributions"
28 python -c "import pyro.infer"
29 python -c "import pyro.optim"
30 '';
31
32 meta = {
33 description = "A Python library for probabilistic modeling and inference";
34 homepage = http://pyro.ai;
35 license = lib.licenses.mit;
36 maintainers = with lib.maintainers; [ teh ];
37 broken = true;
38 };
39}