tjh.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, graphviz
5, jupyter
6, matplotlib
7, networkx
8, opt-einsum
9, pandas
10, pillow
11, pyro-api
12, pythonOlder
13, torch
14, scikit-learn
15, seaborn
16, torchvision
17, tqdm
18, wget
19}:
20
21buildPythonPackage rec {
22 pname = "pyro-ppl";
23 version = "1.8.4";
24 format = "setuptools";
25
26 disabled = pythonOlder "3.7";
27
28 src = fetchPypi {
29 inherit version pname;
30 hash = "sha256-dm+tYeUt9IiF3pbUEhPaH46MG3ns8witUxifzRXBy0E=";
31 };
32
33 propagatedBuildInputs = [
34 pyro-api
35 torch
36 networkx
37 opt-einsum
38 tqdm
39 ];
40
41 passthru.optional-dependencies = {
42 extras = [
43 graphviz
44 jupyter
45 # lap
46 matplotlib
47 pandas
48 pillow
49 scikit-learn
50 seaborn
51 torchvision
52 # visdom
53 wget
54 ];
55 };
56
57 # pyro not shipping tests do simple smoke test instead
58 doCheck = false;
59
60 pythonImportsCheck = [
61 "pyro"
62 "pyro.distributions"
63 "pyro.infer"
64 "pyro.optim"
65 ];
66
67 meta = with lib; {
68 description = "Library for probabilistic modeling and inference";
69 homepage = "http://pyro.ai";
70 license = licenses.asl20;
71 maintainers = with maintainers; [ teh georgewhewell ];
72 };
73}