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

   1 { lib
   2 , buildPythonPackage
   3 , isPy27
   4 , fetchFromGitHub
   5 , pytestCheckHook
   6 , unittest2
   7 , future
   8 , numpy
   9 , pillow
  10 , scipy
  11 , scikitlearn
  12 , scikitimage
  13 , threadpoolctl
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "batchgenerators";
  18   version = "0.21";
  19
  20   disabled = isPy27;
  21
  22   src = fetchFromGitHub {
  23     owner = "MIC-DKFZ";
  24     repo = pname;
  25     rev = "v${version}";
  26     sha256 = "16bk4r0q3m2c9fawpmj4l7kz0x3fyv1spb92grf44gmyricq3jdb";
  27
  28   };
  29
  30   propagatedBuildInputs = [
  31     future numpy pillow scipy scikitlearn scikitimage threadpoolctl
  32   ];
  33
  34   checkInputs = [ pytestCheckHook unittest2 ];
  35
  36   meta = {
  37     description = "2D and 3D image data augmentation for deep learning";
  38     homepage = "https://github.com/MIC-DKFZ/batchgenerators";
  39     license = lib.licenses.asl20;
  40     maintainers = with lib.maintainers; [ bcdarwin ];
  41   };
  42 }