{ lib, buildPythonPackage, pythonOlder, fetchFromGitHub, # build-system setuptools, # dependencies albucore, eval-type-backport, numpy, opencv4, pydantic, pyyaml, scikit-image, scikit-learn, deepdiff, pytestCheckHook, pytest-mock, torch, torchvision, }: buildPythonPackage rec { pname = "albumentations"; version = "1.4.11"; pyproject = true; disabled = pythonOlder "3.8"; src = fetchFromGitHub { owner = "albumentations-team"; repo = "albumentations"; rev = "refs/tags/${version}"; hash = "sha256-1070V9+EZ4qrhxmbMyvTbu89pLoonrn0Peb8nwp2lwA="; }; pythonRemoveDeps = [ "opencv-python" ]; build-system = [ setuptools ]; dependencies = [ albucore eval-type-backport numpy opencv4 pydantic pyyaml scikit-image scikit-learn ]; nativeCheckInputs = [ deepdiff pytestCheckHook pytest-mock torch torchvision ]; disabledTests = [ # this test hangs up "test_transforms" ]; pythonImportsCheck = [ "albumentations" ]; meta = { description = "Fast image augmentation library and easy to use wrapper around other libraries"; homepage = "https://github.com/albumentations-team/albumentations"; changelog = "https://github.com/albumentations-team/albumentations/releases/tag/${version}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ natsukium ]; }; }