experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal contrib changes

Browse Source 
PiperOrigin-RevId: 289918555
Change-Id: Ieeb986b4f35086903e31d87422e61da560ee3fc3
This commit is contained in:
A. Unique TensorFlower 2020-01-15 12:55:58 -08:00 committed by TensorFlower Gardener
parent 3feb0cbaa7
commit 18369ac065
2 changed files with 4 additions and 4 deletions
tensorflow
__init__.py
python/util
module_wrapper.py

4
tensorflow/__init__.py
View File

@ -23,10 +23,6 @@ from __future__ import print_function
# pylint: disable=g-bad-import-order # pylint: disable=g-bad-import-order
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
from tensorflow.python.util.lazy_loader import LazyLoader
contrib = LazyLoader('contrib', globals(), 'tensorflow.contrib')
del LazyLoader
from tensorflow.python.platform import flags # pylint: disable=g-import-not-at-top from tensorflow.python.platform import flags # pylint: disable=g-import-not-at-top
from tensorflow.python.platform import app # pylint: disable=g-import-not-at-top from tensorflow.python.platform import app # pylint: disable=g-import-not-at-top
app.flags = flags app.flags = flags

4
tensorflow/python/util/module_wrapper.py
View File

@ -173,6 +173,8 @@ class TFModuleWrapper(types.ModuleType):
attr_map[name] = attr attr_map[name] = attr
return attr return attr
# Placeholder for Google-internal contrib error
attr = super(TFModuleWrapper, self).__getattribute__(name) attr = super(TFModuleWrapper, self).__getattribute__(name)
# Return and cache dunders and our own members. # Return and cache dunders and our own members.
@ -191,6 +193,8 @@ class TFModuleWrapper(types.ModuleType):
try: try:
attr = getattr(self._tfmw_wrapped_module, name) attr = getattr(self._tfmw_wrapped_module, name)
except AttributeError: except AttributeError:
# Placeholder for Google-internal contrib error
if not self._tfmw_public_apis: if not self._tfmw_public_apis:
raise raise
if name not in self._tfmw_public_apis: if name not in self._tfmw_public_apis: