message: Fix multi_worker_callback_tf2_test test target by only running it with CPU. The test is not using GPU anyway.
diff: @@ -374,14 +374,11 @@ cuda_py_test(
  ],
  )
  -cuda_py_test(
  +py_test(
  name = "multi_worker_callback_tf2_test",
  srcs = ["multi_worker_callback_tf2_test.py"],
  + python_version = "PY3",
  shard_count = 10,
  - tags = [
  - "manual", # TODO(b/148795660): Reenable once the flakiness is fixed.
  - "notap",
  - ],
  deps = [
  "//tensorflow/python/distribute:collective_all_reduce_strategy",
  "//tensorflow/python/distribute:combinations",
  