From 750ceb3938a9cf03d10c9703884abbce43dbc28d Mon Sep 17 00:00:00 2001 From: Michael Banfield Date: Wed, 1 May 2019 12:21:56 -0700 Subject: [PATCH] Disable discovery cache for TPUClusterResolver. PiperOrigin-RevId: 246176508 --- .../distribute/cluster_resolver/tpu_cluster_resolver.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py b/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py index 6d0e3b8d4d2..c6c9a19c483 100644 --- a/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py +++ b/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py @@ -108,13 +108,11 @@ class TPUClusterResolver(ClusterResolver): if self._discovery_url: return discovery.build( - 'tpu', 'v1alpha1', - credentials=credentials, - discoveryServiceUrl=self._discovery_url) + 'tpu', 'v1alpha1', credentials=credentials, + discoveryServiceUrl=self._discovery_url, cache_discovery=False) else: return discovery.build( - 'tpu', 'v1alpha1', - credentials=credentials) + 'tpu', 'v1alpha1', credentials=credentials, cache_discovery=False) def _request_compute_metadata(self, path): req = Request('%s/computeMetadata/v1/%s' % (_GCE_METADATA_ENDPOINT, path),