tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5
6 # build-system
7 hatchling,
8
9 # dependencies
10 colorama,
11 onnx,
12 packaging,
13 sympy,
14}:
15
16buildPythonPackage (finalAttrs: {
17 pname = "onnxslim";
18 version = "0.1.82";
19 pyproject = true;
20
21 src = fetchFromGitHub {
22 owner = "inisis";
23 repo = "OnnxSlim";
24 tag = "v${finalAttrs.version}";
25 hash = "sha256-hrrCodLaHVo/YRq0HczxogcZQSwZKxZthyLYxz/+XJ0=";
26 };
27
28 build-system = [
29 hatchling
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 };
49})