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 , torchvision
  17 , tqdm
  18 , wget
  19 }:
  20
  21 buildPythonPackage rec {
  22   pname = "pyro-ppl";
  23   version = "1.8.2";
  24   format = "setuptools";
  25
  26   disabled = pythonOlder "3.7";
  27
  28   src = fetchPypi {
  29     inherit version pname;
  30     hash = "sha256-4Afl0ROCpY78+4+61xxy6vEGbIZsaNxURXDEGMGCiks=";
  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 }