diff --git a/pkgs/development/python-modules/pymatting/01-kdtree-signature.patch b/pkgs/development/python-modules/pymatting/01-kdtree-signature.patch new file mode 100644 index 000000000000..2193b260aea1 --- /dev/null +++ b/pkgs/development/python-modules/pymatting/01-kdtree-signature.patch @@ -0,0 +1,13 @@ +diff --git a/tests/test_kdtree.py b/tests/test_kdtree.py +index 0110083..049fa07 100644 +--- a/tests/test_kdtree.py ++++ b/tests/test_kdtree.py +@@ -23,7 +23,7 @@ def run_kdtree(): + + tree = cKDTree(data_points) + +- distances2, indices2 = tree.query(query_points, k=k, n_jobs=-1) ++ distances2, indices2 = tree.query(query_points, k=k) + + t2 = time.perf_counter() + diff --git a/pkgs/development/python-modules/pymatting/default.nix b/pkgs/development/python-modules/pymatting/default.nix new file mode 100644 index 000000000000..0d10efd616a4 --- /dev/null +++ b/pkgs/development/python-modules/pymatting/default.nix @@ -0,0 +1,55 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, numba +, numpy +, pillow +, scipy +, pytestCheckHook +, +}: +buildPythonPackage rec { + pname = "pymatting"; + version = "1.1.2"; + format = "setuptools"; + + src = fetchFromGitHub { + owner = "pymatting"; + repo = "pymatting"; + rev = "v${version}"; + hash = "sha256-9eRpsWwXAkp6aw1ZWJsUFf0BMIN0UBFc2rW1lltL2cw="; + }; + + patches = [ ./01-kdtree-signature.patch ]; + + propagatedBuildInputs = [ + numba + numpy + pillow + scipy + ]; + + pythonImportsCheck = [ "pymatting" ]; + + nativeCheckInputs = [ + pytestCheckHook + ]; + + disabledTests = [ + # no access to input data set + # see: https://github.com/pymatting/pymatting/blob/master/tests/download_images.py + "test_alpha" + "test_laplacians" + "test_preconditioners" + "test_lkm" + ]; + + meta = with lib; { + description = "A Python library for alpha matting"; + homepage = "https://github.com/pymatting/pymatting"; + changelog = "https://github.com/pymatting/pymatting/blob/${src.rev}/CHANGELOG.md"; + license = licenses.mit; + maintainers = with maintainers; [ blaggacao ]; + }; +} + diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 98fc8165e010..6123e85a21e8 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -9047,6 +9047,8 @@ self: super: with self; { pymatgen = callPackage ../development/python-modules/pymatgen { }; + pymatting = callPackage ../development/python-modules/pymatting { }; + pymaven-patch = callPackage ../development/python-modules/pymaven-patch { }; pymavlink = callPackage ../development/python-modules/pymavlink { };