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

Merge pull request #39525 from tensorflow/mm-fix-major-api-version

Browse Source 
Fix tests when `tf._major_api_version` does not exist
This commit is contained in:
Mihai Maruseac 2020-05-14 02:17:30 +00:00 committed by GitHub
commit 3ffdb91f12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/tools/api/tests/module_test.py
View File

@ -74,7 +74,8 @@ class ModuleTest(test.TestCase):
tf.summary.image tf.summary.image
# If we use v2 API, check for create_file_writer, # If we use v2 API, check for create_file_writer,
# otherwise check for FileWriter. # otherwise check for FileWriter.
if hasattr(tf, '_major_api_version') and tf._major_api_version == 2: if hasattr(tf, '_major_api_version'):
if tf._major_api_version == 2:
tf.summary.create_file_writer tf.summary.create_file_writer
else: else:
tf.summary.FileWriter tf.summary.FileWriter
@ -82,6 +83,7 @@ class ModuleTest(test.TestCase):
def testInternalKerasImport(self): def testInternalKerasImport(self):
normalization_parent = layers.BatchNormalization.__module__.split('.')[-1] normalization_parent = layers.BatchNormalization.__module__.split('.')[-1]
if hasattr(tf, '_major_api_version'):
if tf._major_api_version == 2: if tf._major_api_version == 2:
self.assertEqual('normalization_v2', normalization_parent) self.assertEqual('normalization_v2', normalization_parent)
self.assertTrue(layers.BatchNormalization._USE_V2_BEHAVIOR) self.assertTrue(layers.BatchNormalization._USE_V2_BEHAVIOR)