From 9b30b99383295e70d13afd5c0e59fd8c29ac92a1 Mon Sep 17 00:00:00 2001 From: alsaiduq-lab Date: Sun, 16 Mar 2025 13:38:47 -0600 Subject: [PATCH] added --- pkgs/default.nix | 14 +++++++++++--- pkgs/python-pymatting/default.nix | 28 ++++++++++++++++++++++++++++ pkgs/python-rembg/default.nix | 3 ++- 3 files changed, 41 insertions(+), 4 deletions(-) create mode 100644 pkgs/python-pymatting/default.nix diff --git a/pkgs/default.nix b/pkgs/default.nix index 45ce300..4836427 100644 --- a/pkgs/default.nix +++ b/pkgs/default.nix @@ -1,9 +1,17 @@ { pkgs ? import {}, lib ? pkgs.lib }: -{ - fish-rust = pkgs.callPackage ./fish-rust {}; - python-rembg = pkgs.callPackage ./python-rembg { +let + python-pymatting = pkgs.callPackage ./python-pymatting { inherit (pkgs) lib fetchPypi; python310Packages = pkgs.python310.pkgs; }; +in +{ + fish-rust = pkgs.callPackage ./fish-rust {}; + python-pymatting = python-pymatting; + python-rembg = pkgs.callPackage ./python-rembg { + inherit (pkgs) lib fetchPypi; + python310Packages = pkgs.python310.pkgs; + inherit python-pymatting; + }; } diff --git a/pkgs/python-pymatting/default.nix b/pkgs/python-pymatting/default.nix new file mode 100644 index 0000000..32f88af --- /dev/null +++ b/pkgs/python-pymatting/default.nix @@ -0,0 +1,28 @@ +{ lib, python310Packages, fetchPypi }: + +python310Packages.buildPythonPackage rec { + pname = "pymatting"; + version = "1.1.13"; + format = "setuptools"; + + src = fetchPypi { + inherit pname version; + sha256 = "sha256-LNt8S++s3e9Netwt6ONKJy3mOdYIrlwKCGE9+kJTgQE="; # Updated hash + }; + + propagatedBuildInputs = with python310Packages; [ + numpy + scipy + pillow + numba + ]; + + doCheck = false; + + meta = with lib; { + description = "A library for alpha matting"; + homepage = "https://github.com/pymatting/pymatting"; + license = licenses.mit; + platforms = platforms.all; + }; +} diff --git a/pkgs/python-rembg/default.nix b/pkgs/python-rembg/default.nix index 5f56cec..d2c009b 100644 --- a/pkgs/python-rembg/default.nix +++ b/pkgs/python-rembg/default.nix @@ -1,4 +1,4 @@ -{ lib, python310Packages, fetchPypi }: +{ lib, python310Packages, fetchPypi, python-pymatting }: python310Packages.buildPythonPackage rec { pname = "rembg"; @@ -27,6 +27,7 @@ python310Packages.buildPythonPackage rec { filetype pooch pympler + python-pymatting scikit-image scipy tqdm