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

Update TF OSS API generation script to support OSS keras PIP package.

Browse Source 
PiperOrigin-RevId: 359374348
Change-Id: I17b63aea5086250a19b98ad04ad9d17ded3d1f88
This commit is contained in:
Scott Zhu 2021-02-24 14:44:15 -08:00 committed by TensorFlower Gardener
parent e30cc0bfb0
commit c023a82916
4 changed files with 64 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
tensorflow/api_template.__init__.py
View File

@ -84,13 +84,22 @@ if _module_dir:
_current_module.__path__ = [_module_dir] + _current_module.__path__ _current_module.__path__ = [_module_dir] + _current_module.__path__
setattr(_current_module, "estimator", estimator) setattr(_current_module, "estimator", estimator)
try: if _os.environ.get("_PREFER_OSS_KERAS", False):
from .python.keras.api._v2 import keras try:
_current_module.__path__ = ( from keras.api._v2 import keras
[_module_util.get_parent_dir(keras)] + _current_module.__path__) _current_module.__path__ = (
setattr(_current_module, "keras", keras) [_module_util.get_parent_dir(keras)] + _current_module.__path__)
except ImportError: setattr(_current_module, "keras", keras)
pass except ImportError:
pass
else:
try:
from .python.keras.api._v2 import keras
_current_module.__path__ = (
[_module_util.get_parent_dir(keras)] + _current_module.__path__)
setattr(_current_module, "keras", keras)
except ImportError:
pass
# Explicitly import lazy-loaded modules to support autocompletion. # Explicitly import lazy-loaded modules to support autocompletion.
# pylint: disable=g-import-not-at-top # pylint: disable=g-import-not-at-top

23
tensorflow/api_template_v1.__init__.py
View File

@ -75,13 +75,22 @@ if _module_dir:
_current_module.__path__ = [_module_dir] + _current_module.__path__ _current_module.__path__ = [_module_dir] + _current_module.__path__
setattr(_current_module, "estimator", estimator) setattr(_current_module, "estimator", estimator)
try: if _os.environ.get("_PREFER_OSS_KERAS", False):
from .python.keras.api._v1 import keras try:
_current_module.__path__ = ( from keras.api._v1 import keras
[_module_util.get_parent_dir(keras)] + _current_module.__path__) _current_module.__path__ = (
setattr(_current_module, "keras", keras) [_module_util.get_parent_dir(keras)] + _current_module.__path__)
except ImportError: setattr(_current_module, "keras", keras)
pass except ImportError:
pass
else:
try:
from .python.keras.api._v1 import keras
_current_module.__path__ = (
[_module_util.get_parent_dir(keras)] + _current_module.__path__)
setattr(_current_module, "keras", keras)
except ImportError:
pass
# Explicitly import lazy-loaded modules to support autocompletion. # Explicitly import lazy-loaded modules to support autocompletion.
# pylint: disable=g-import-not-at-top # pylint: disable=g-import-not-at-top

23
tensorflow/compat_template.__init__.py
View File

@ -52,13 +52,22 @@ if _module_dir:
_current_module.__path__ = [_module_dir] + _current_module.__path__ _current_module.__path__ = [_module_dir] + _current_module.__path__
setattr(_current_module, "estimator", estimator) setattr(_current_module, "estimator", estimator)
try: if _os.environ.get("_PREFER_OSS_KERAS", False):
from tensorflow.python.keras.api._v2 import keras try:
_current_module.__path__ = ( from keras.api._v2 import keras
[_module_util.get_parent_dir(keras)] + _current_module.__path__) _current_module.__path__ = (
setattr(_current_module, "keras", keras) [_module_util.get_parent_dir(keras)] + _current_module.__path__)
except ImportError: setattr(_current_module, "keras", keras)
pass except ImportError:
pass
else:
try:
from tensorflow.python.keras.api._v2 import keras
_current_module.__path__ = (
[_module_util.get_parent_dir(keras)] + _current_module.__path__)
setattr(_current_module, "keras", keras)
except ImportError:
pass
# Explicitly import lazy-loaded modules to support autocompletion. # Explicitly import lazy-loaded modules to support autocompletion.
# pylint: disable=g-import-not-at-top # pylint: disable=g-import-not-at-top

23
tensorflow/compat_template_v1.__init__.py
View File

@ -42,13 +42,22 @@ if _module_dir:
_current_module.__path__ = [_module_dir] + _current_module.__path__ _current_module.__path__ = [_module_dir] + _current_module.__path__
setattr(_current_module, "estimator", estimator) setattr(_current_module, "estimator", estimator)
try: if _os.environ.get("_PREFER_OSS_KERAS", False):
from tensorflow.python.keras.api._v1 import keras try:
_current_module.__path__ = ( from keras.api._v1 import keras
[_module_util.get_parent_dir(keras)] + _current_module.__path__) _current_module.__path__ = (
setattr(_current_module, "keras", keras) [_module_util.get_parent_dir(keras)] + _current_module.__path__)
except ImportError: setattr(_current_module, "keras", keras)
pass except ImportError:
pass
else:
try:
from tensorflow.python.keras.api._v1 import keras
_current_module.__path__ = (
[_module_util.get_parent_dir(keras)] + _current_module.__path__)
setattr(_current_module, "keras", keras)
except ImportError:
pass
# Explicitly import lazy-loaded modules to support autocompletion. # Explicitly import lazy-loaded modules to support autocompletion.
# pylint: disable=g-import-not-at-top # pylint: disable=g-import-not-at-top