Internal change.

PiperOrigin-RevId: 236942537

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,

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"):