diff --git a/pkgs/development/python-modules/paddlex/default.nix b/pkgs/development/python-modules/paddlex/default.nix new file mode 100644 index 000000000000..68f196c3dcf1 --- /dev/null +++ b/pkgs/development/python-modules/paddlex/default.nix @@ -0,0 +1,95 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + setuptools, + numpy, + pillow, + pyyaml, + chardet, + colorlog, + filelock, + pandas, + prettytable, + py-cpuinfo, + pydantic, + requests, + ruamel-yaml, + typing-extensions, + ujson, +}: + +let + gputil = buildPythonPackage rec { + pname = "gputil"; + version = "1.4.0"; + pyproject = true; + + src = fetchFromGitHub { + owner = "anderskm"; + repo = "gputil"; + tag = "v${version}"; + hash = "sha256-iOyB653BMmDBtK1fM1ZyddjlnaypsuLMOV0sKaBt+yE="; + }; + + build-system = [ setuptools ]; + + meta = { + homepage = "https://github.com/anderskm/gputil"; + license = lib.licenses.mit; + description = "Getting GPU status from NVIDA GPUs using nvidia-smi"; + changelog = "https://github.com/anderskm/gputil/releases/tag/${src.tag}"; + }; + }; +in +buildPythonPackage rec { + pname = "paddlex"; + version = "3.0.1"; + pyproject = true; + + src = fetchFromGitHub { + owner = "PaddlePaddle"; + repo = "PaddleX"; + tag = "v${version}"; + hash = "sha256-qov5nqGIsSfaho2dcWVsyWKQlJsIJgdX3rDz66JtLDI="; + }; + + build-system = [ setuptools ]; + + pythonRelaxDeps = [ + "numpy" + "pandas" + ]; + + dependencies = [ + chardet + colorlog + filelock + numpy + pandas + pillow + prettytable + py-cpuinfo + pydantic + pyyaml + requests + ruamel-yaml + typing-extensions + ujson + gputil + ]; + + meta = { + homepage = "https://github.com/PaddlePaddle/PaddleX"; + license = lib.licenses.asl20; + description = "All-in-One Development Tool based on PaddlePaddle"; + changelog = "https://github.com/PaddlePaddle/PaddleX/releases/tag/${src.tag}"; + maintainers = with lib.maintainers; [ emaryn ]; + platforms = [ + "x86_64-linux" + "aarch64-linux" + "x86_64-darwin" + "aarch64-darwin" + ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 6aac78daecee..3223d97f9607 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -10912,6 +10912,8 @@ self: super: with self; { paddlepaddle = callPackage ../development/python-modules/paddlepaddle { }; + paddlex = callPackage ../development/python-modules/paddlex { }; + pagelabels = callPackage ../development/python-modules/pagelabels { }; paginate = callPackage ../development/python-modules/paginate { };