push sheeet

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

@@ -0,0 +1,95 @@
+{
+  lib,
+  buildPythonPackage,
+  pythonOlder,
+  fetchFromGitHub,
+
+  # build-system
+  setuptools,
+
+  # dependencies
+  albucore,
+  numpy,
+  opencv-python,
+  pydantic,
+  pyyaml,
+  scipy,
+
+  # optional dependencies
+  huggingface-hub,
+  pillow,
+  torch,
+
+  # tests
+  deepdiff,
+  pytestCheckHook,
+  pytest-mock,
+  scikit-image,
+  scikit-learn,
+  torchvision,
+}:
+
+buildPythonPackage rec {
+  pname = "albumentations";
+  version = "2.0.8";
+  pyproject = true;
+
+  disabled = pythonOlder "3.9";
+
+  src = fetchFromGitHub {
+    owner = "albumentations-team";
+    repo = "albumentations";
+    tag = version;
+    hash = "sha256-8vUipdkIelRtKwMw63oUBDN/GUI0gegMGQaqDyXAOTQ=";
+  };
+
+  patches = [
+    ./dont-check-for-updates.patch
+  ];
+
+  pythonRelaxDeps = [ "opencv-python" ];
+
+  build-system = [ setuptools ];
+
+  dependencies = [
+    albucore
+    numpy
+    opencv-python
+    pydantic
+    pyyaml
+    scipy
+  ];
+
+  optional-dependencies = {
+    hub = [ huggingface-hub ];
+    pytorch = [ torch ];
+    text = [ pillow ];
+  };
+
+  nativeCheckInputs = [
+    deepdiff
+    pytestCheckHook
+    pytest-mock
+    scikit-image
+    scikit-learn
+    torch
+    torchvision
+  ];
+
+  disabledTests = [
+    "test_pca_inverse_transform"
+    # these tests hang
+    "test_keypoint_remap_methods"
+    "test_multiprocessing_support"
+  ];
+
+  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/${src.tag}";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ natsukium ];
+  };
+}

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

@@ -0,0 +1,12 @@
+diff --git a/albumentations/__init__.py b/albumentations/__init__.py
+index 44ee9b9..ea3bc50 100644
+--- a/albumentations/__init__.py
++++ b/albumentations/__init__.py
+@@ -22,7 +22,3 @@ from .core.transforms_interface import *
+ 
+ with suppress(ImportError):
+     from .pytorch import *
+-
+-# Perform the version check after all other initializations
+-if os.getenv("NO_ALBUMENTATIONS_UPDATE", "").lower() not in {"true", "1"}:
+-    check_for_updates()