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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , setuptools
   5 , numpy
   6 , opencv4
   7 , pyyaml
   8 , qudida
   9 , scikit-image
  10 , scipy
  11 , deepdiff
  12 , pytestCheckHook
  13 , pythonOlder
  14 , pythonRelaxDepsHook
  15 , torch
  16 , torchvision
  17 }:
  18
  19 buildPythonPackage rec {
  20   pname = "albumentations";
  21   version = "1.4.2";
  22   pyproject = true;
  23
  24   disabled = pythonOlder "3.8";
  25
  26   src = fetchPypi {
  27     inherit pname version;
  28     hash = "sha256-lznWLJocXdfwnhAZ33V5ZdlFCAsNa0u/rjfkjmHBQOg=";
  29   };
  30
  31   nativeBuildInputs = [
  32     pythonRelaxDepsHook
  33   ];
  34
  35   pythonRemoveDeps = [
  36     "opencv-python"
  37   ];
  38
  39   build-system = [
  40     setuptools
  41   ];
  42
  43   dependencies = [
  44     numpy
  45     opencv4
  46     pyyaml
  47     qudida
  48     scikit-image
  49     scipy
  50   ];
  51
  52   nativeCheckInputs = [
  53     deepdiff
  54     pytestCheckHook
  55     torch
  56     torchvision
  57   ];
  58
  59   disabledTests = [
  60     # this test hangs up
  61     "test_transforms"
  62   ];
  63
  64   pythonImportsCheck = [ "albumentations" ];
  65
  66   meta = with lib; {
  67     description = "Fast image augmentation library and easy to use wrapper around other libraries";
  68     homepage = "https://github.com/albumentations-team/albumentations";
  69     changelog = "https://github.com/albumentations-team/albumentations/releases/tag/${version}";
  70     license = licenses.mit;
  71     maintainers = with maintainers; [ natsukium ];
  72   };
  73 }