message: TF API Fixit: Remove private symbol IndependentWorkerTestBase by converting worker_training_state_test to TF2/eager test.
diff: @@ -19,7 +19,6 @@ import sys
  from absl.testing import parameterized
  from tensorflow.python.distribute import combinations as ds_combinations
  -from tensorflow.python.distribute import multi_worker_test_base as test_base
  from tensorflow.python.framework import test_combinations as combinations
  from tensorflow.python.keras import callbacks
  from tensorflow.python.keras.distribute import multi_worker_testing_utils
  @@ -27,17 +26,14 @@ from tensorflow.python.lib.io import file_io
  from tensorflow.python.platform import test
  -class ModelCheckpointTest(test_base.IndependentWorkerTestBase,
  - parameterized.TestCase):
  +class ModelCheckpointTest(test.TestCase, parameterized.TestCase):
  @ds_combinations.generate(
  combinations.combine(
  - mode=['graph'],
  - required_gpus=[0, 1],
  + mode=['eager'],
  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()
  @@ -47,13 +43,11 @@ class ModelCheckpointTest(test_base.IndependentWorkerTestBase,
  filepath=saving_filepath, save_weights_only=save_weights_only)
  ]
  self.assertFalse(file_io.file_exists_v2(saving_filepath))
  - model.fit(
  - x=train_ds, epochs=2, steps_per_epoch=2, callbacks=callbacks_list)
  + model.fit(x=train_ds, epochs=2, steps_per_epoch=2, callbacks=callbacks_list)
  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)
  + 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)
  if __name__ == '__main__':
  