From 18d86872ff0708709745bcb368279a7f14ccb310 Mon Sep 17 00:00:00 2001
From: Chuanhao Zhuge <chuanhao@google.com>
Date: Tue, 9 Feb 2021 13:30:55 -0800
Subject: [PATCH] Disable a few more Python tests that TFRT doesn't support.

PiperOrigin-RevId: 356578473
Change-Id: I5eaece2460d130a66ea7045e9f98697df1066538
---
 tensorflow/python/BUILD                                    | 5 ++++-
 tensorflow/python/keras/callbacks_test.py                  | 2 ++
 tensorflow/python/keras/engine/BUILD                       | 1 +
 tensorflow/python/keras/engine/training_dataset_test.py    | 2 ++
 tensorflow/python/kernel_tests/control_flow_ops_py_test.py | 1 +
 5 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index 64f1cf634d0..654aced8453 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -3066,7 +3066,10 @@ tf_py_test(
     size = "small",
     srcs = ["ops/collective_ops_test.py"],
     python_version = "PY3",
-    tags = ["no_rocm"],
+    tags = [
+        "no_rocm",
+        "no_tfrt",  # TODO(b/179692150)
+    ],
     deps = [
         ":client_testlib",
         ":collective_ops",
diff --git a/tensorflow/python/keras/callbacks_test.py b/tensorflow/python/keras/callbacks_test.py
index e5df8ad8d26..193b8cb2b32 100644
--- a/tensorflow/python/keras/callbacks_test.py
+++ b/tensorflow/python/keras/callbacks_test.py
@@ -40,6 +40,7 @@ from tensorflow.python.data.ops import readers
 from tensorflow.python.eager import context
 from tensorflow.python.framework import ops
 from tensorflow.python.framework import random_seed
+from tensorflow.python.framework import test_util
 from tensorflow.python.keras import keras_parameterized
 from tensorflow.python.keras import testing_utils
 from tensorflow.python.keras.engine import sequential
@@ -1598,6 +1599,7 @@ class KerasCallbacksTest(keras_parameterized.TestCase):
             callbacks=cbks,
             epochs=1)
 
+  @test_util.disable_tfrt('b/179690526')
   def test_progbar_infers_steps(self):
     x, y = np.ones((10, 1)), np.ones((10, 1))
     data = dataset_ops.DatasetV2.from_tensor_slices((x, y)).batch(2)
diff --git a/tensorflow/python/keras/engine/BUILD b/tensorflow/python/keras/engine/BUILD
index 1a5b0a92e8c..a4e12e99443 100644
--- a/tensorflow/python/keras/engine/BUILD
+++ b/tensorflow/python/keras/engine/BUILD
@@ -274,6 +274,7 @@ tf_py_test(
     shard_count = 4,
     tags = [
         "no_oss_py38",  # TODO(b/150615192)
+        "no_tfrt",  # TODO(b/179805675)
         "nomac",  # TODO(mihaimaruseac): b/127695564
     ],
     deps = [
diff --git a/tensorflow/python/keras/engine/training_dataset_test.py b/tensorflow/python/keras/engine/training_dataset_test.py
index 987c2ebfd7e..242022ac173 100644
--- a/tensorflow/python/keras/engine/training_dataset_test.py
+++ b/tensorflow/python/keras/engine/training_dataset_test.py
@@ -27,6 +27,7 @@ from tensorflow.python import keras
 from tensorflow.python.data.experimental.ops import cardinality
 from tensorflow.python.data.ops import dataset_ops
 from tensorflow.python.framework import ops
+from tensorflow.python.framework import test_util
 from tensorflow.python.keras import callbacks
 from tensorflow.python.keras import keras_parameterized
 from tensorflow.python.keras import metrics as metrics_module
@@ -398,6 +399,7 @@ class TestTrainingWithDataset(keras_parameterized.TestCase):
 
   @keras_parameterized.run_with_all_model_types
   @keras_parameterized.run_all_keras_modes(always_skip_v1=True)
+  @test_util.disable_tfrt('b/179459136')
   def test_finite_dataset_unknown_cardinality_no_step_with_train_and_val(self):
 
     class CaptureStdout(object):
diff --git a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py
index 54bbd2b2e9e..9026100077d 100644
--- a/tensorflow/python/kernel_tests/control_flow_ops_py_test.py
+++ b/tensorflow/python/kernel_tests/control_flow_ops_py_test.py
@@ -1460,6 +1460,7 @@ class ControlFlowTest(test.TestCase, parameterized.TestCase):
 
 
   @test_util.run_in_graph_and_eager_modes
+  @test_util.disable_tfrt("b/179459136")
   def testWhileAutoControlDeps(self):
     # Legacy while_loop fails this test because it produces deprecation notices
     # in stderr.