Remove pywrap_tensorflow from compat.v1.

PiperOrigin-RevId: 238562893
This commit is contained in:
Anna R 2019-03-14 18:34:08 -07:00 committed by TensorFlower Gardener
parent fea13a48df
commit 6d3d026ca0
2 changed files with 6 additions and 3 deletions

View File

@ -21,7 +21,6 @@ from __future__ import print_function as _print_function
import os as _os
# pylint: disable=g-bad-import-order
from tensorflow.python import pywrap_tensorflow # pylint: disable=unused-import
# API IMPORTS PLACEHOLDER

View File

@ -373,8 +373,12 @@ class ApiCompatibilityTest(test.TestCase):
golden_file_pattern = os.path.join(
resource_loader.get_root_dir_with_all_resources(),
_KeyToFilePath('*', api_version))
self._checkBackwardsCompatibility(tf.compat.v1, golden_file_pattern,
api_version)
omit_golden_symbols_map = {'tensorflow': 'pywrap_tensorflow'}
self._checkBackwardsCompatibility(
tf.compat.v1,
golden_file_pattern,
api_version,
omit_golden_symbols_map=omit_golden_symbols_map)
def testAPIBackwardsCompatibilityV2(self):
api_version = 2