提交 d7fd2485 编写于 作者: R Rick Chao 提交者: TensorFlower Gardener

Add tf.estimator as a valid v1 api prefix in testAllAPIV1 check in

tf_upgrade_v2_test.py.

PiperOrigin-RevId: 224624681
上级 8ca891f5
......@@ -198,6 +198,7 @@ class TestUpgrade(test_util.TensorFlowTestCase):
_, _, _, text = self._upgrade("tf." + name)
if (text and
not text.startswith("tf.compat.v1") and
not text.startswith("tf.estimator") and
text not in v1_symbols):
self.assertFalse(
True, "Symbol %s generated from %s not in v1 API" % (
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册