tjh.dev
1{ buildPythonPackage, fetchPypi, lib, pytorch, contextlib2
2, graphviz, networkx, six, opt-einsum, tqdm }:
3buildPythonPackage rec {
4 version = "0.3.1.post1";
5 pname = "pyro-ppl";
6
7 src = fetchPypi {
8 inherit version pname;
9 sha256 = "f5174bcd1b74fb455db948413dcbf5ac1a13dbddc6c54c6d215dd2e4b6b9548d";
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 };
38}