commit 000f6a6729016825ae76ae0963c55aef2a23b538 · pyrox.dev/nixpkgs

+10 -9

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

···

       9
       9
        
       

     

       10
       10
        
         # dependencies

     

       11
       11
        
         albucore,

     

       12
       12
       -
         eval-type-backport,

     

       13
       12
        
         numpy,

     

       14
       13
        
         opencv-python,

     

       15
       14
        
         pydantic,

     

       16
       15
        
         pyyaml,

     

       17
       17
       -
         scikit-image,

     

       18
       16
        
         scipy,

     

       19
       17
        
       

     

       20
       18
        
         # optional dependencies

     

       21
       19
        
         huggingface-hub,

     

       22
       20
        
         pillow,

     

       21
       21
       +
         torch,

     

       23
       22
        
       

     

       24
       23
        
         # tests

     

       25
       24
        
         deepdiff,

     

       26
       25
        
         pytestCheckHook,

     

       27
       26
        
         pytest-mock,

     

       28
       28
       -
         torch,

     

       27
       27
       +
         scikit-image,

     

       28
       28
       +
         scikit-learn,

     

       29
       29
        
         torchvision,

     

       30
       30
        
       }:

     

       31
       31
        
       

     

       32
       32
        
       buildPythonPackage rec {

     

       33
       33
        
         pname = "albumentations";

     

       34
       34
       -
         version = "2.0.0";

     

       34
       34
       +
         version = "2.0.5";

     

       35
       35
        
         pyproject = true;

     

       36
       36
        
       

     

       37
       37
        
         disabled = pythonOlder "3.9";

     
···

       40
       40
        
           owner = "albumentations-team";

     

       41
       41
        
           repo = "albumentations";

     

       42
       42
        
           tag = version;

     

       43
       43
       -
           hash = "sha256-8WEOI2J2H4PNhyb9LoIUMofGKx9AHPiPddkQCSdh8/A=";

     

       43
       43
       +
           hash = "sha256-WqU25I1DxBqZAXd2+sNMUv/HOL4towlGTnFnpCGmMgY=";

     

       44
       44
        
         };

     

       45
       45
        
       

     

       46
       46
        
         patches = [

     
···

       53
       53
        
       

     

       54
       54
        
         dependencies = [

     

       55
       55
        
           albucore

     

       56
       56
       -
           eval-type-backport

     

       57
       56
        
           numpy

     

       58
       57
        
           opencv-python

     

       59
       58
        
           pydantic

     

       60
       59
        
           pyyaml

     

       61
       61
       -
           scikit-image

     

       62
       60
        
           scipy

     

       63
       61
        
         ];

     

       64
       62
        
       

     

       65
       63
        
         optional-dependencies = {

     

       66
       64
        
           hub = [ huggingface-hub ];

     

       65
       65
       +
           pytorch = [ torch ];

     

       67
       66
        
           text = [ pillow ];

     

       68
       67
        
         };

     

       69
       68
        
       

     
···

       71
       70
        
           deepdiff

     

       72
       71
        
           pytestCheckHook

     

       73
       72
        
           pytest-mock

     

       73
       73
       +
           scikit-image

     

       74
       74
       +
           scikit-learn

     

       74
       75
        
           torch

     

       75
       76
        
           torchvision

     

       76
       77
        
         ];

     

       77
       78
        
       

     

       78
       79
        
         disabledTests = [

     

       79
       80
        
           "test_pca_inverse_transform"

     

       80
       80
       -
           # this test hangs up

     

       81
       81
       -
           "test_transforms"

     

       81
       81
       +
           # test hangs

     

       82
       82
       +
           "test_keypoint_remap_methods"

     

       82
       83
        
         ];

     

       83
       84
        
       

     

       84
       85
        
         pythonImportsCheck = [ "albumentations" ];

+5 -5

pkgs/development/python-modules/albumentations/dont-check-for-updates.patch

···

       1
       1
        
       diff --git a/albumentations/__init__.py b/albumentations/__init__.py

     

       2
       2
       -
       index 0b3b531..7c69c65 100644

     

       2
       2
       +
       index 44ee9b9..ea3bc50 100644

     

       3
       3
        
       --- a/albumentations/__init__.py

     

       4
       4
        
       +++ b/albumentations/__init__.py

     

       5
       5
       -
       @@ -7,7 +7,3 @@ from .augmentations import *

     

       6
       6
       -
        from .core.composition import *

     

       7
       7
       -
        from .core.serialization import *

     

       8
       8
       -
        from .core.transforms_interface import *

     

       5
       5
       +
       @@ -22,7 +22,3 @@ from .core.transforms_interface import *

     

       6
       6
       +
        

     

       7
       7
       +
        with suppress(ImportError):

     

       8
       8
       +
            from .pytorch import *

     

       9
       9
        
       -

     

       10
       10
        
       -# Perform the version check after all other initializations

     

       11
       11
        
       -if os.getenv("NO_ALBUMENTATIONS_UPDATE", "").lower() not in {"true", "1"}: