diff --git a/tensorflow/python/keras/distribute/worker_training_state_test.py b/tensorflow/python/keras/distribute/worker_training_state_test.py index cb65747c239..67f15b9748a 100644 --- a/tensorflow/python/keras/distribute/worker_training_state_test.py +++ b/tensorflow/python/keras/distribute/worker_training_state_test.py @@ -24,7 +24,6 @@ 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.framework.errors_impl import NotFoundError from tensorflow.python.keras import callbacks from tensorflow.python.keras.distribute import multi_worker_testing_utils from tensorflow.python.lib.io import file_io @@ -51,13 +50,8 @@ class ModelCheckpointTest(test_base.IndependentWorkerTestBase, filepath=saving_filepath, save_weights_only=save_weights_only) ] self.assertFalse(file_io.file_exists_v2(saving_filepath)) - - try: - model.fit( - x=train_ds, epochs=2, steps_per_epoch=2, callbacks=callbacks_list) - except NotFoundError as e: - if 'Failed to create a NewWriteableFile' in e.message: - self.skipTest('b/138941852, path not found error in Windows py35.') + 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')