tjh.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, numpy
5, packaging
6, protobuf
7, onnx
8, unittestCheckHook
9, onnxruntime
10}:
11
12buildPythonPackage rec {
13 pname = "onnxconverter-common";
14 version = "1.13.0";
15
16 format = "setuptools";
17
18 src = fetchFromGitHub {
19 owner = "microsoft";
20 repo = "onnxconverter-common";
21 rev = "refs/tags/v${version}";
22 hash = "sha256-VT9ly0d0Yhw1J6C521oUyaCx4WtFSdpyk8EdIKlre3c=";
23 };
24
25 propagatedBuildInputs = [
26 numpy
27 packaging
28 protobuf
29 onnx
30 ];
31
32 nativeCheckInputs = [
33 onnxruntime
34 unittestCheckHook
35 ];
36
37 unittestFlagsArray = [ "-s" "tests" ];
38
39 # Failing tests
40 # https://github.com/microsoft/onnxconverter-common/issues/242
41 doCheck = false;
42
43 meta = {
44 description = "ONNX Converter and Optimization Tools";
45 homepage = "https://github.com/microsoft/onnxconverter-common";
46 changelog = "https://github.com/microsoft/onnxconverter-common/releases/tag/v${version}";
47 maintainers = with lib.maintainers; [ fridh ];
48 license = with lib.licenses; [ mit ];
49 };
50}