pkgs/development/python-modules/pyro-ppl/default.nix

   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 , setuptools
  17 , torchvision
  18 , tqdm
  19 , wget
  20 }:
  21
  22 buildPythonPackage rec {
  23   pname = "pyro-ppl";
  24   version = "1.9.0";
  25   pyproject = true;
  26
  27   disabled = pythonOlder "3.8";
  28
  29   src = fetchPypi {
  30     inherit version pname;
  31     hash = "sha256-QfTABRWVaCgPvFEWSJYKmKKxpBACfYvQpDIgrJsQLN8=";
  32   };
  33
  34   nativeBuildInputs = [
  35     setuptools
  36   ];
  37
  38   propagatedBuildInputs = [
  39     pyro-api
  40     torch
  41     networkx
  42     opt-einsum
  43     tqdm
  44   ];
  45
  46   passthru.optional-dependencies = {
  47     extras = [
  48       graphviz
  49       jupyter
  50       # lap
  51       matplotlib
  52       pandas
  53       pillow
  54       scikit-learn
  55       seaborn
  56       torchvision
  57       # visdom
  58       wget
  59     ];
  60   };
  61
  62   # pyro not shipping tests do simple smoke test instead
  63   doCheck = false;
  64
  65   pythonImportsCheck = [
  66     "pyro"
  67     "pyro.distributions"
  68     "pyro.infer"
  69     "pyro.optim"
  70   ];
  71
  72   meta = with lib; {
  73     description = "Library for probabilistic modeling and inference";
  74     homepage = "http://pyro.ai";
  75     changelog = "https://github.com/pyro-ppl/pyro/releases/tag/${version}";
  76     license = licenses.asl20;
  77     maintainers = with maintainers; [ teh georgewhewell ];
  78   };
  79 }