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