From b8ad028165d0ee441e20e264c83dcb82201da530 Mon Sep 17 00:00:00 2001 From: Shining Sun Date: Wed, 19 Jun 2019 11:45:13 -0700 Subject: [PATCH] Remove obsolete TODOs. PiperOrigin-RevId: 254039929 --- .../python/keras/distribute/multi_worker_callback_test.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tensorflow/python/keras/distribute/multi_worker_callback_test.py b/tensorflow/python/keras/distribute/multi_worker_callback_test.py index 3f88d69147e..cc44afa15b0 100644 --- a/tensorflow/python/keras/distribute/multi_worker_callback_test.py +++ b/tensorflow/python/keras/distribute/multi_worker_callback_test.py @@ -146,16 +146,12 @@ class KerasMultiWorkerCallbackTest(test_base.IndependentWorkerTestBase, **kwargs): extension = os.path.splitext(saving_filepath)[1] - # TODO(rchao): Remove using .h5 once b/134551335 is fixed. - extension = '.h5' # Incorporate type/index information and thread id in saving_filepath to # ensure every worker has a unique path. Note that in normal use case the # saving_filepath will be the same for all workers, but we use different # ones here just to test out chief saves checkpoint but non-chief doesn't. - # TODO(b/134551335): Must save to hdf5 until bug with copying - # MirroredVariables is resolved. saving_filepath = os.path.join( test_obj.get_temp_dir(), 'checkpoint_%s_%d%s' % (test_base.get_task_type(), test_base.get_task_index(), extension))