From 35d0488ab84ec73b1af8b107e73f463931b34e41 Mon Sep 17 00:00:00 2001 From: Anna R Date: Tue, 5 Mar 2019 16:24:40 -0800 Subject: [PATCH] Internal change. PiperOrigin-RevId: 236942537 --- tensorflow/tools/api/tests/api_compatibility_test.py | 1 + tensorflow/tools/compatibility/tf_upgrade_v2_test.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tensorflow/tools/api/tests/api_compatibility_test.py b/tensorflow/tools/api/tests/api_compatibility_test.py index fe52a2bde7d..0c1f9de1a36 100644 --- a/tensorflow/tools/api/tests/api_compatibility_test.py +++ b/tensorflow/tools/api/tests/api_compatibility_test.py @@ -295,6 +295,7 @@ class ApiCompatibilityTest(test.TestCase): visitor.do_not_descend_map['tf'].append('contrib') if FLAGS.only_test_core_api: visitor.do_not_descend_map['tf'].extend(_NON_CORE_PACKAGES) + visitor.private_map['tf.compat'] = ['v1', 'v2'] traverse.traverse(tf.compat.v2, visitor) def _checkBackwardsCompatibility(self, diff --git a/tensorflow/tools/compatibility/tf_upgrade_v2_test.py b/tensorflow/tools/compatibility/tf_upgrade_v2_test.py index c2eefde1e9b..0c465807240 100644 --- a/tensorflow/tools/compatibility/tf_upgrade_v2_test.py +++ b/tensorflow/tools/compatibility/tf_upgrade_v2_test.py @@ -97,6 +97,7 @@ class TestUpgrade(test_util.TensorFlowTestCase): cls.v2_symbols["tf." + name] = attr visitor = public_api.PublicAPIVisitor(symbol_collector) + visitor.private_map["tf.compat"] = ["v1"] traverse.traverse(tf.compat.v2, visitor) if hasattr(tf.compat, "v1"):