pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5
6 # build-system
7 setuptools,
8
9 # dependencies
10 colorama,
11 onnx,
12 packaging,
13 sympy,
14}:
15
16buildPythonPackage rec {
17 pname = "onnxslim";
18 version = "0.1.69";
19 pyproject = true;
20
21 src = fetchFromGitHub {
22 owner = "inisis";
23 repo = "OnnxSlim";
24 tag = "v${version}";
25 hash = "sha256-UwuZDAPTwIVJTseu8iIO2Q8OOaFcwiwLS/auWOpIpnQ=";
26 };
27
28 build-system = [
29 setuptools
30 ];
31
32 dependencies = [
33 colorama
34 onnx
35 packaging
36 sympy
37 ];
38
39 pythonImportsCheck = [ "onnxslim" ];
40
41 # __main__.py: error: the following arguments are required: --model-dir
42 doCheck = false;
43
44 meta = {
45 description = "Toolkit to Help Optimize Onnx Model";
46 homepage = "https://pypi.org/project/onnxslim/";
47 license = lib.licenses.mit;
48 maintainers = with lib.maintainers; [ ferrine ];
49 };
50}