From d0106f72ea1f5a859dcd7a93674fce2a54890b07 Mon Sep 17 00:00:00 2001
From: Mihai Maruseac <mihaimaruseac@google.com>
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__':