pyrox.dev
python310Packages.tf2onnx: init at 1.14.0
authored by happysalada and committed by Yt da5286cd 78386f6b
+86
pkgs/development/python-modules/tf2onnx/default.nix
+86
pkgs/development/python-modules/tf2onnx/default.nix
···
1
+
{ lib
2
+
, buildPythonPackage
3
+
, fetchFromGitHub
4
+
, pythonRelaxDepsHook
5
+
, pytest-runner
6
+
# runtime dependencies
7
+
, numpy
8
+
, onnx
9
+
, requests
10
+
, six
11
+
, flatbuffers
12
+
, protobuf
13
+
, tensorflow
14
+
# check dependencies
15
+
, pytestCheckHook
16
+
, graphviz
17
+
, parameterized
18
+
, pytest-cov
19
+
, pyyaml
20
+
, timeout-decorator
21
+
, onnxruntime
22
+
, keras
23
+
}:
24
+
25
+
buildPythonPackage rec {
26
+
pname = "tf2onnx";
27
+
version = "1.14.0";
28
+
format = "setuptools";
29
+
30
+
src = fetchFromGitHub {
31
+
owner = "onnx";
32
+
repo = "tensorflow-onnx";
33
+
rev = "v${version}";
34
+
hash = "sha256-JpXwf+GLjn0krsb5KnEhVuemWa0V2+wF10neDsdtOfI=";
35
+
};
36
+
37
+
nativeBuildInputs = [
38
+
pythonRelaxDepsHook
39
+
pytest-runner
40
+
];
41
+
42
+
pythonRelaxDeps = [
43
+
"flatbuffers"
44
+
];
45
+
46
+
propagatedBuildInputs = [
47
+
numpy
48
+
onnx
49
+
requests
50
+
six
51
+
flatbuffers
52
+
protobuf
53
+
tensorflow
54
+
onnxruntime
55
+
];
56
+
57
+
pythonImportsCheck = [ "tf2onnx" ];
58
+
59
+
nativeCheckInputs = [
60
+
pytestCheckHook
61
+
graphviz
62
+
parameterized
63
+
pytest-cov
64
+
pyyaml
65
+
timeout-decorator
66
+
keras
67
+
];
68
+
69
+
# TODO investigate the failures
70
+
disabledTestPaths = [
71
+
"tests/test_backend.py"
72
+
"tests/test_einsum_helper.py"
73
+
"tests/test_einsum_optimizers.py"
74
+
];
75
+
76
+
disabledTests = [
77
+
"test_profile_conversion_time"
78
+
];
79
+
80
+
meta = with lib; {
81
+
description = "Convert TensorFlow, Keras, Tensorflow.js and Tflite models to ONNX";
82
+
homepage = "https://github.com/onnx/tensorflow-onnx";
83
+
license = licenses.asl20;
84
+
maintainers = with maintainers; [ happysalada ];
85
+
};
86
+
}
+3
-2
pkgs/development/python-modules/transformers/default.nix
+3
-2
pkgs/development/python-modules/transformers/default.nix
···
18
18
, scikit-learn
19
19
, tensorflow
20
20
, onnxconverter-common
21
+
, tf2onnx
21
22
, torch
22
23
, accelerate
23
24
, faiss
···
99
100
tf = [
100
101
tensorflow
101
102
onnxconverter-common
102
-
# tf2onnx
103
+
tf2onnx
103
104
# tensorflow-text
104
105
# keras-nlp
105
106
];
···
119
120
];
120
121
onnx = [
121
122
onnxconverter-common
122
-
# tf2onnx
123
+
tf2onnx
123
124
onnxruntime
124
125
];
125
126
modelcreation = [
+2
pkgs/top-level/python-packages.nix
+2
pkgs/top-level/python-packages.nix
···
12387
12387
12388
12388
textx = callPackage ../development/python-modules/textx { };
12389
12389
12390
+
tf2onnx = callPackage ../development/python-modules/tf2onnx { };
12391
+
12390
12392
tflearn = callPackage ../development/python-modules/tflearn { };
12391
12393
12392
12394
tftpy = callPackage ../development/python-modules/tftpy { };