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

Fix tests when tf._major_api_version does not exist

Browse Source
This commit is contained in:
Mihai Maruseac 2020-05-13 18:33:18 -07:00
parent 90d1854a80
commit d0106f72ea
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -74,20 +74,22 @@ class ModuleTest(test.TestCase):
tf.summary.image
# If we use v2 API, check for create_file_writer,
# otherwise check for FileWriter.
if hasattr(tf, '_major_api_version') and tf._major_api_version == 2:
tf.summary.create_file_writer
else:
tf.summary.FileWriter
if hasattr(tf, '_major_api_version'):
if tf._major_api_version == 2:
tf.summary.create_file_writer
else:
tf.summary.FileWriter
# pylint: enable=pointless-statement
def testInternalKerasImport(self):
normalization_parent = layers.BatchNormalization.__module__.split('.')[-1]
if tf._major_api_version == 2:
self.assertEqual('normalization_v2', normalization_parent)
self.assertTrue(layers.BatchNormalization._USE_V2_BEHAVIOR)
else:
self.assertEqual('normalization', normalization_parent)
self.assertFalse(layers.BatchNormalization._USE_V2_BEHAVIOR)
if hasattr(tf, '_major_api_version'):
if tf._major_api_version == 2:
self.assertEqual('normalization_v2', normalization_parent)
self.assertTrue(layers.BatchNormalization._USE_V2_BEHAVIOR)
else:
self.assertEqual('normalization', normalization_parent)
self.assertFalse(layers.BatchNormalization._USE_V2_BEHAVIOR)
if __name__ == '__main__':