Disable a few more Python tests that TFRT doesn't support.

PiperOrigin-RevId: 356578473
Change-Id: I5eaece2460d130a66ea7045e9f98697df1066538

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",

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)

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 = [

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

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.