From d0106f72ea1f5a859dcd7a93674fce2a54890b07 Mon Sep 17 00:00:00 2001 From: Mihai Maruseac Date: Wed, 13 May 2020 18:33:18 -0700 Subject: [PATCH] Fix tests when `tf._major_api_version` does not exist --- tensorflow/tools/api/tests/module_test.py | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/tensorflow/tools/api/tests/module_test.py b/tensorflow/tools/api/tests/module_test.py index 692ce94d006..c0870e6c134 100644 --- a/tensorflow/tools/api/tests/module_test.py +++ b/tensorflow/tools/api/tests/module_test.py @@ -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__':