Unverified Commit 43cf50bd authored by Martin Weinelt's avatar Martin Weinelt Committed by GitHub
Browse files

Merge pull request #310959 from TomaSajt/albumentations 

python312Packages.albumentations: fix dependencies, fix tests
parents 8625a10f 11d25281

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

+14 −7
Original line number Diff line number Diff line 
{ lib

, buildPythonPackage

, fetchPypi

, fetchFromGitHub

, setuptools

, deepdiff

, numpy

, opencv4

, pyyaml

, qudida

, scikit-image

, scikit-learn

, scipy

, deepdiff

, pydantic

, pytestCheckHook

, pythonOlder

, pythonRelaxDepsHook

, torch

, torchvision

, typing-extensions

}:



buildPythonPackage rec {
@@ -23,9 +25,11 @@ buildPythonPackage rec { 


  disabled = pythonOlder "3.8";



  src = fetchPypi {

    inherit pname version;

    hash = "sha256-VNpClOBdoaQOqU92Mm/Z5Q7P+fZzR+m43SFA9pSi1M0=";

  src = fetchFromGitHub {

    owner = "albumentations-team";

    repo = "albumentations";

    rev = "refs/tags/${version}";

    hash = "sha256-7t1+22zzFtkZaAyOo6xjk+MXT9N44PmQ/NRRfvLeRVk=";

  };



  nativeBuildInputs = [
@@ -34,6 +38,7 @@ buildPythonPackage rec { 


  pythonRemoveDeps = [

    "opencv-python"

    "pydantic"

  ];



  build-system = [
@@ -43,10 +48,12 @@ buildPythonPackage rec { 
  dependencies = [

    numpy

    opencv4

    pydantic

    pyyaml

    qudida

    scikit-image

    scikit-learn

    scipy

    typing-extensions

  ];



  nativeCheckInputs = [