message: Fixing documentation error
diff: # See the License for the specific language governing permissions and
  # limitations under the License.
  # ==============================================================================
  -"""Tests for TFCONFIGClusterResolver."""
  +"""Tests for SageMakerClusterResolver."""
  from __future__ import absolute_import
  from __future__ import division
  