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