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

Swap out modules in py2 mode in a cleaner fashion.

Browse Source 
PiperOrigin-RevId: 288526813
Change-Id: I86efd4d804c0c873856307cf4a969270eb7bbae8
This commit is contained in:
Dan Moldovan 2020-01-07 10:41:39 -08:00 committed by TensorFlower Gardener
parent 1bf56f31cc
commit b00d66ebe0
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/autograph/utils/compat_util.py
View File

@ -20,7 +20,6 @@ from __future__ import print_function
import importlib import importlib
import sys import sys
import types
import six import six
@ -36,10 +35,4 @@ def deprecated_py2_support(module_name):
"""Swaps calling module with a Py2-specific implementation. Noop in Py3.""" """Swaps calling module with a Py2-specific implementation. Noop in Py3."""
if six.PY2: if six.PY2:
legacy_module = importlib.import_module(module_name + '_deprecated_py2') legacy_module = importlib.import_module(module_name + '_deprecated_py2')
current_module = sys.modules[module_name] sys.modules[module_name] = legacy_module
for name, target_val in legacy_module.__dict__.items():
if isinstance(target_val, types.FunctionType):
replacement = types.FunctionType(
target_val.__code__, current_module.__dict__, target_val.__name__,
target_val.__defaults__, target_val.__closure__)
current_module.__dict__[name] = replacement