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