message: Address Ubuntu Sanity error
diff: @@ -320,9 +320,10 @@ class TPUStrategyV2(distribute_lib.Strategy):
  """Synchronous training in TPU donuts or Pods.
  Args:
  - tpu_cluster_resolver: A `tf.distribute.cluster_resolver.TPUClusterResolver`
  - instance, which provides information about the TPU cluster. If None, it
  - will assume running on a local TPU worker.
  + tpu_cluster_resolver: A
  + `tf.distribute.cluster_resolver.TPUClusterResolver` instance, which
  + provides information about the TPU cluster. If None, it will assume
  + running on a local TPU worker.
  experimental_device_assignment: Optional
  `tf.tpu.experimental.DeviceAssignment` to specify the placement of
  replicas on the TPU cluster.
  