message: Fix failed tests in v1 graph mode for model checkpoint in OSS.
diff: @@ -41,6 +41,7 @@ class ModelCheckpointTest(test_base.IndependentWorkerTestBase,
  file_format=['h5', 'tf'],
  save_weights_only=[True, False]))
  def testCheckpointExists(self, file_format, save_weights_only):
  + with self.cached_session():
  train_ds, _ = multi_worker_testing_utils.mnist_synthetic_dataset(64, 2)
  model = multi_worker_testing_utils.get_mnist_model((28, 28, 1))
  saving_dir = self.get_temp_dir()
  @@ -60,7 +61,8 @@ class ModelCheckpointTest(test_base.IndependentWorkerTestBase,
  tf_saved_model_exists = file_io.file_exists_v2(saving_filepath)
  tf_weights_only_checkpoint_exists = file_io.file_exists_v2(
  saving_filepath + '.index')
  - self.assertTrue(tf_saved_model_exists or tf_weights_only_checkpoint_exists)
  + self.assertTrue(
  + tf_saved_model_exists or tf_weights_only_checkpoint_exists)
  if __name__ == '__main__':
  