OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 12:20:24 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Normalize Protocol Goldens Generation

Browse Source 
PiperOrigin-RevId: 403171391
Change-Id: I486aa30488c6fa90a56fa3939cc2bb9bb277ff80
This commit is contained in:
Faizan Muhammad 2021-10-14 13:21:11 -07:00 committed by TensorFlower Gardener
parent c3da9e2f00
commit ac8b31f8c2
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/tools/api/lib/python_object_to_proto_visitor.py
View File

@ -97,11 +97,17 @@ _NORMALIZE_TYPE[(
'tensorflow.python.framework.ops.Tensor')] = (
"<class 'tensorflow.python.framework.ops.Tensor'>")
_NORMALIZE_TYPE['typing.Generic'] = "<class 'typing.Generic'>"
# TODO(mdan): Remove once the golden files are generated in Python 3.7.
# TODO(b/203104448): Remove once the golden files are generated in Python 3.7.
_NORMALIZE_TYPE["<class 'typing._GenericAlias'>"] = 'typing.Union'
# TODO(mdan): Remove once the golden files are generated in Python 3.9.
# TODO(b/203104448): Remove once the golden files are generated in Python 3.9.
_NORMALIZE_TYPE["<class 'typing._UnionGenericAlias'>"] = 'typing.Union'
# TODO(b/203104448): Remove once the golden files are generated in Python 3.8.
_NORMALIZE_TYPE[
"<class 'typing_extensions._ProtocolMeta'>"] = ("<class "
"'typing._ProtocolMeta'>")
# TODO(b/203104448): Remove once the golden files are generated in Python 3.8.
_NORMALIZE_TYPE[
"<class 'typing_extensions.Protocol'>"] = "<class 'typing.Protocol'>"
if sys.version_info.major == 3 and sys.version_info.minor >= 8:
_NORMALIZE_TYPE["<class '_collections._tuplegetter'>"] = "<type 'property'>"