message: Temporarily disable tensorflow/python/keras:distribute_strategy_test_tpu test target for flakiness.
diff: # Contains the Keras API (internal TensorFlow version).
  load("//tensorflow:tensorflow.bzl", "tf_py_test")
  load("//tensorflow:tensorflow.bzl", "cuda_py_test")
  -load("//tensorflow/core:platform/default/distribute.bzl", "distribute_py_test")
  licenses(["notice"]) # Apache 2.0
  @@ -1426,10 +1425,13 @@ py_library(
  ],
  )
  -distribute_py_test(
  +# TODO(b/129155953): Fix the flakiness and re-enable distribute_strategy_test_tpu target.
  +cuda_py_test(
  name = "distribute_strategy_test",
  srcs = ["distribute_strategy_test.py"],
  - full_precision = True,
  + additional_deps = [
  + ":distribute_strategy_test_lib",
  + ],
  main = "distribute_strategy_test.py",
  shard_count = 32,
  tags = [
  @@ -1438,7 +1440,4 @@ distribute_py_test(
  "no_windows_gpu",
  "notsan",
  ],
  - deps = [
  - ":distribute_strategy_test_lib",
  - ],
  )
  