Skip to content

Commit e886d45

Browse files
hwangdeyuguschmue
andcommitted
add commit value to onnx file, remove expired code and add License in generating version.py
Signed-off-by: Deyu Huang <[email protected]> Co-authored-by: Guenther Schmuelling <[email protected]>
1 parent 8e43a17 commit e886d45

File tree

7 files changed

+13
-14
lines changed

7 files changed

+13
-14
lines changed

setup.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,9 @@ def finalize_options(self):
3737
pass
3838

3939
def run(self):
40+
alv2_license = "# SPDX-License-Identifier: Apache-2.0\n\n"
4041
with open(os.path.join(SRC_DIR, 'version.py'), 'w') as f:
41-
f.write(dedent('''
42+
f.write(alv2_license + dedent('''
4243
version = '{version}'
4344
git_version = '{git_version}'
4445
'''.format(**dict(VersionInfo._asdict()))))

tests/keras2onnx_applications/model_source/densenet_1/densenet_1.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,6 @@
88
- [Densely Connected Convolutional Networks](https://arxiv.org/pdf/1608.06993.pdf)
99
- [The One Hundred Layers Tiramisu: Fully Convolutional DenseNets for Semantic Segmentation](https://arxiv.org/pdf/1611.09326.pdf)
1010
'''
11-
from __future__ import print_function
12-
from __future__ import absolute_import
13-
from __future__ import division
1411

1512
import warnings
1613

tests/keras2onnx_applications/model_source/densenet_1/subpixel.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,6 @@
33
# From https://github.com/titu1994/DenseNet/blob/master/subpixel.py
44
# Modifications Copyright (c) Microsoft.
55

6-
from __future__ import absolute_import
7-
86
from mock_keras2onnx.proto import keras
97
from keras import backend as K
108
from keras.engine import Layer

tf2onnx/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
"tfonnx", "shape_inference", "schemas", "tf_utils", "tf_loader", "convert"]
77

88
import onnx
9-
from .version import version as __version__
9+
from .version import git_version, version as __version__
1010
from . import verbose_logging as logging
1111
from tf2onnx import tfonnx, utils, graph, graph_builder, graph_matcher, shape_inference, schemas, convert # pylint: disable=wrong-import-order
12-
#from tf2onnx import tf_utils, tf_loader

tf2onnx/graph.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import numpy as np
1313

1414
from onnx import helper, numpy_helper, shape_inference, AttributeProto, TensorProto
15-
from tf2onnx import utils, __version__
15+
from tf2onnx import utils, __version__, git_version
1616
from tf2onnx.utils import make_name, port_name, find_opset
1717
from tf2onnx import optimizer
1818
from tf2onnx.schemas import get_schema, infer_onnx_shape_dtype
@@ -1189,9 +1189,10 @@ def make_model(self, graph_doc, optimize=False, graph_name="tf2onnx", external_t
11891189
graph = self.make_graph(graph_doc, graph_name, external_tensor_storage)
11901190

11911191
if "producer_name" not in kwargs:
1192-
kwargs = {"producer_name": "tf2onnx",
1193-
"producer_version": __version__}
1194-
1192+
kwargs = {
1193+
"producer_name": "tf2onnx",
1194+
"producer_version": __version__ + " " + git_version[:6]
1195+
}
11951196
if "opset_imports" not in kwargs:
11961197
opsets = [helper.make_opsetid(constants.ONNX_DOMAIN, self._opset)]
11971198
opsets.append(constants.AI_ONNX_ML_OPSET)
@@ -1668,6 +1669,8 @@ def get_onnx_model_properties(onnx_model_proto):
16681669
kwargs["producer_name"] = onnx_model_proto.producer_name
16691670
if onnx_model_proto.HasField('producer_version'):
16701671
kwargs["producer_version"] = onnx_model_proto.producer_version
1672+
if onnx_model_proto.HasField('producer_git_version'):
1673+
kwargs["producer_git_version"] = onnx_model_proto.producer_git_version
16711674
if onnx_model_proto.HasField('domain'):
16721675
kwargs["domain"] = onnx_model_proto.domain
16731676
if onnx_model_proto.HasField('model_version'):

tf2onnx/tfonnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -411,7 +411,7 @@ def process_tf_graph(tf_graph, continue_on_error=False, verbose=False, target=No
411411

412412
opset = utils.find_opset(opset)
413413
logger.info("Using tensorflow=%s, onnx=%s, tf2onnx=%s/%s",
414-
get_tf_version(), utils.get_onnx_version(), tf2onnx.__version__, tf2onnx.version.git_version[:6])
414+
get_tf_version(), utils.get_onnx_version(), tf2onnx.__version__, tf2onnx.git_version[:6])
415415
logger.info("Using opset <onnx, %s>", opset)
416416
if opset > schemas.get_max_supported_opset_version():
417417
logger.warning("Currently installed onnx package %s is too low to support opset %s, "

tf2onnx/version.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
# SPDX-License-Identifier: Apache-2.0
22

3+
34
version = '1.10.0'
4-
git_version = '219e00c073f6e73fba7335630dcf1f96cc82c983'
5+
git_version = '8e43a1783a1921a4f5f20d51ae9b48926706a939'

0 commit comments

Comments
 (0)