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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   pythonOlder,
   5   fetchFromGitHub,
   6
   7   # build-system
   8   setuptools,
   9
  10   # dependencies
  11   albucore,
  12   eval-type-backport,
  13   numpy,
  14   opencv-python,
  15   pydantic,
  16   pyyaml,
  17   scikit-image,
  18   scipy,
  19
  20   # optional dependencies
  21   huggingface-hub,
  22   pillow,
  23
  24   # tests
  25   deepdiff,
  26   pytestCheckHook,
  27   pytest-mock,
  28   torch,
  29   torchvision,
  30 }:
  31
  32 buildPythonPackage rec {
  33   pname = "albumentations";
  34   version = "1.4.22";
  35   pyproject = true;
  36
  37   disabled = pythonOlder "3.9";
  38
  39   src = fetchFromGitHub {
  40     owner = "albumentations-team";
  41     repo = "albumentations";
  42     rev = "refs/tags/${version}";
  43     hash = "sha256-kRf8LhRWtzGnhPrQo5aT/4a2sNQCdwAmFFzwcE0QnxM=";
  44   };
  45
  46   patches = [
  47     ./dont-check-for-updates.patch
  48   ];
  49
  50   pythonRelaxDeps = [ "opencv-python" ];
  51
  52   build-system = [ setuptools ];
  53
  54   dependencies = [
  55     albucore
  56     eval-type-backport
  57     numpy
  58     opencv-python
  59     pydantic
  60     pyyaml
  61     scikit-image
  62     scipy
  63   ];
  64
  65   optional-dependencies = {
  66     hub = [ huggingface-hub ];
  67     text = [ pillow ];
  68   };
  69
  70   nativeCheckInputs = [
  71     deepdiff
  72     pytestCheckHook
  73     pytest-mock
  74     torch
  75     torchvision
  76   ];
  77
  78   disabledTests = [
  79     "test_pca_inverse_transform"
  80     # this test hangs up
  81     "test_transforms"
  82   ];
  83
  84   pythonImportsCheck = [ "albumentations" ];
  85
  86   meta = {
  87     description = "Fast image augmentation library and easy to use wrapper around other libraries";
  88     homepage = "https://github.com/albumentations-team/albumentations";
  89     changelog = "https://github.com/albumentations-team/albumentations/releases/tag/${version}";
  90     license = lib.licenses.mit;
  91     maintainers = with lib.maintainers; [ natsukium ];
  92   };
  93 }