From c64097cb5f68c28491fd6e2b954d203b2fb5eca5 Mon Sep 17 00:00:00 2001 From: Rick Chao Date: Fri, 22 May 2020 10:43:42 -0700 Subject: [PATCH] Re-enable multi_worker_tutorial_test (by fixing the timeout on shorter epoch/steps) PiperOrigin-RevId: 312842049 Change-Id: If0d7f0fcb4463c718f5532f62cca17ac23cab99a --- tensorflow/python/keras/distribute/BUILD | 1 - .../python/keras/distribute/multi_worker_tutorial_test.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index b7fe3b5bda6..6a39ebc5007 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -432,7 +432,6 @@ py_test( tags = [ "noasan", # TODO(b/156029134) "nomsan", # TODO(b/156029134) - "notap", # TODO(b/157253858) "notsan", # TODO(b/156029134) ], deps = [ diff --git a/tensorflow/python/keras/distribute/multi_worker_tutorial_test.py b/tensorflow/python/keras/distribute/multi_worker_tutorial_test.py index 1a46bcd7499..3f9ab18f89c 100644 --- a/tensorflow/python/keras/distribute/multi_worker_tutorial_test.py +++ b/tensorflow/python/keras/distribute/multi_worker_tutorial_test.py @@ -120,8 +120,8 @@ class MultiWorkerTutorialTest(parameterized.TestCase, test.TestCase): multi_worker_model.fit( multi_worker_dataset, - epochs=3, - steps_per_epoch=70, + epochs=2, + steps_per_epoch=20, callbacks=callbacks) with test_util.skip_if_error(self, errors_impl.UnavailableError):