Andrew Audibert 209ede00c5 [tf.data service] Fix timeouts in distributed_epoch tests.
This fixes an issue where tests would get stuck retrying GetSplit requests to the dispatcher after the test had already shut down the dispatcher. Now the retry period is configurable, so that tests can set it to 1 second instead of the default 1 hour.

PiperOrigin-RevId: 337935450
Change-Id: I11bdfba4e731af314fc0016ee5b113dbc339cbdd
2020-10-19 14:44:44 -07:00
..
2020-09-30 16:03:47 -07:00