message: Temporarily skip MultiWorkerTrainingStateTest.testCheckpointExists test when saving to saved_model when tf2.enabled().
diff: @@ -19,8 +19,8 @@ from __future__ import division
  from __future__ import print_function
  import os
  import sys
  -import tempfile
  from absl.testing import parameterized
  +from tensorflow.python import tf2
  from tensorflow.python.distribute import combinations
  from tensorflow.python.distribute import multi_worker_test_base as test_base
  from tensorflow.python.keras import callbacks
  @@ -41,13 +41,17 @@ class MultiWorkerTrainingStateTest(test_base.IndependentWorkerTestBase,
  def testCheckpointExists(self, file_format, save_weights_only):
  train_ds, _ = multi_worker_testing_utils.mnist_synthetic_dataset(64, 2)
  model = multi_worker_testing_utils.get_mnist_model((28, 28, 1))
  - saving_dir = tempfile.mkdtemp()
  + saving_dir = self.get_temp_dir()
  saving_filepath = os.path.join(saving_dir, 'checkpoint.' + file_format)
  callbacks_list = [
  callbacks.ModelCheckpoint(
  filepath=saving_filepath, save_weights_only=save_weights_only)
  ]
  self.assertFalse(training_state.checkpoint_exists(saving_filepath))
  +
  + if file_format == 'tf' and not save_weights_only and tf2.enabled():
  + self.skipTest('b/135217014')
  +
  model.fit(x=train_ds, epochs=2, steps_per_epoch=2, callbacks=callbacks_list)
  self.assertTrue(training_state.checkpoint_exists(saving_filepath))
  self.assertTrue(
  