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