tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 setuptools,
6 numpy,
7 pillow,
8 pyyaml,
9 chardet,
10 colorlog,
11 filelock,
12 pandas,
13 prettytable,
14 py-cpuinfo,
15 pydantic,
16 requests,
17 ruamel-yaml,
18 typing-extensions,
19 ujson,
20 distutils,
21 huggingface-hub,
22 nix-update-script,
23}:
24
25let
26 gputil = buildPythonPackage rec {
27 pname = "gputil";
28 version = "1.4.0";
29 pyproject = true;
30
31 src = fetchFromGitHub {
32 owner = "anderskm";
33 repo = "gputil";
34 tag = "v${version}";
35 hash = "sha256-iOyB653BMmDBtK1fM1ZyddjlnaypsuLMOV0sKaBt+yE=";
36 };
37
38 build-system = [ setuptools ];
39
40 dependencies = [ distutils ];
41
42 pythonImportsCheck = [ "GPUtil" ];
43
44 meta = {
45 homepage = "https://github.com/anderskm/gputil";
46 license = lib.licenses.mit;
47 description = "Getting GPU status from NVIDA GPUs using nvidia-smi";
48 changelog = "https://github.com/anderskm/gputil/releases/tag/${src.tag}";
49 };
50 };
51in
52buildPythonPackage rec {
53 pname = "paddlex";
54 version = "3.1.4";
55 pyproject = true;
56
57 src = fetchFromGitHub {
58 owner = "PaddlePaddle";
59 repo = "PaddleX";
60 tag = "v${version}";
61 hash = "sha256-Oc8fgAv8T/9PjxW8yU31t3m3CUxFuAXdVS71BGhtlJo=";
62 };
63
64 build-system = [ setuptools ];
65
66 pythonRelaxDeps = [
67 "numpy"
68 "pandas"
69 ];
70
71 dependencies = [
72 chardet
73 colorlog
74 filelock
75 numpy
76 pandas
77 pillow
78 prettytable
79 py-cpuinfo
80 pydantic
81 pyyaml
82 requests
83 ruamel-yaml
84 typing-extensions
85 ujson
86 gputil
87 huggingface-hub
88 ];
89
90 passthru.updateScript = nix-update-script { };
91
92 meta = {
93 homepage = "https://github.com/PaddlePaddle/PaddleX";
94 license = lib.licenses.asl20;
95 description = "All-in-One Development Tool based on PaddlePaddle";
96 changelog = "https://github.com/PaddlePaddle/PaddleX/releases/tag/${src.tag}";
97 maintainers = [ ];
98 platforms = [
99 "x86_64-linux"
100 "aarch64-linux"
101 "x86_64-darwin"
102 "aarch64-darwin"
103 ];
104 };
105}