pkgs/development/python-modules/pywick/default.nix

   1 { buildPythonPackage
   2 , fetchFromGitHub
   3 , pythonOlder
   4 , pytest
   5 , h5py
   6 , hickle
   7 , numpy
   8 , pandas
   9 , pillow
  10 , six
  11 , pytorch
  12 , torchvision
  13 , tqdm
  14 , lib
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname   = "pywick";
  19   version = "0.5.6";
  20
  21   disabled = pythonOlder "3.6";
  22
  23   src = fetchFromGitHub {
  24     owner = "achaiah";
  25     repo = pname;
  26     rev = "v${version}";
  27     sha256 = "1gmlifnv9kji0d1jwg1pa8d96zg48w17qg0sgxwy1y1jf3hn37bm";
  28   };
  29
  30   propagatedBuildInputs = [
  31     h5py hickle numpy pandas pillow six pytorch torchvision tqdm
  32   ];
  33
  34   checkInputs = [ pytest ];
  35
  36   checkPhase = ''
  37     runHook preCheck
  38     pytest tests/
  39     runHook postCheck
  40   '';
  41
  42   meta = {
  43     description = "High-level training framework for Pytorch";
  44     homepage = "https://github.com/achaiah/pywick";
  45     license = lib.licenses.mit;
  46     maintainers = with lib.maintainers; [ bcdarwin ];
  47   };
  48 }