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

   1 { lib
   2 , buildPythonPackage
   3 , pythonOlder
   4 , fetchFromGitHub
   5 , pytestCheckHook
   6 , future
   7 , numpy
   8 , pillow
   9 , scipy
  10 , scikit-learn
  11 , scikit-image
  12 , threadpoolctl
  13 }:
  14
  15 buildPythonPackage rec {
  16   pname = "batchgenerators";
  17   version = "0.25";
  18   format = "setuptools";
  19
  20   disabled = pythonOlder "3.7";
  21
  22   src = fetchFromGitHub {
  23     owner = "MIC-DKFZ";
  24     repo = pname;
  25     rev = "v${version}";
  26     hash = "sha256-L2mWH2t8PN9o1M67KDdl1Tj2ZZ02MY4icsJY2VNrj3A=";
  27   };
  28
  29   propagatedBuildInputs = [
  30     future
  31     numpy
  32     pillow
  33     scipy
  34     scikit-learn
  35     scikit-image
  36     threadpoolctl
  37   ];
  38
  39   # see https://github.com/MIC-DKFZ/batchgenerators/pull/78
  40   postPatch = ''
  41     substituteInPlace setup.py \
  42       --replace '"unittest2",' ""
  43   '';
  44
  45   nativeCheckInputs = [
  46     pytestCheckHook
  47   ];
  48
  49   # see https://github.com/MIC-DKFZ/batchgenerators/pull/78
  50   disabledTestPaths = [ "tests/test_axis_mirroring.py" ];
  51
  52   pythonImportsCheck = [
  53     "batchgenerators"
  54     "batchgenerators.augmentations"
  55     "batchgenerators.dataloading"
  56     "batchgenerators.datasets"
  57     "batchgenerators.transforms"
  58     "batchgenerators.utilities"
  59   ];
  60
  61   meta = with lib; {
  62     description = "2D and 3D image data augmentation for deep learning";
  63     homepage = "https://github.com/MIC-DKFZ/batchgenerators";
  64     license = licenses.asl20;
  65     maintainers = with maintainers; [ bcdarwin ];
  66   };
  67 }