message: PSv2: Increase the shard count for parameter server training test to avoid a timeout and unblock tap.
diff: @@ -860,7 +860,7 @@ distribute_py_test(
  name = "parameter_server_training_test",
  srcs = ["parameter_server_training_test.py"],
  python_version = "PY3",
  - shard_count = 1,
  + shard_count = 4, # TODO(b/184290570): Investigate why only 1 shard times out.
  tags = [
  "multi_and_single_gpu",
  "no_tfrt", # TODO(b/180537361): Reenable TFRT after the issue is resolved.
  