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),