From c2d12f9be6802c7d4dd2ba0d8417b6b520f25730 Mon Sep 17 00:00:00 2001 From: Dwight J Lyle Date: Mon, 10 Jun 2019 15:39:31 -0700 Subject: [PATCH] Point tpu_cluster_resolver.py to the v1 TPU API. The v1 API is available. It might be good to consider allowing customers to pick which API version they want, but for now use the v1 API. --- .../distribute/cluster_resolver/tpu_cluster_resolver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py b/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py index f41571ba94d5e8..551347ebd8fd4a 100644 --- a/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py +++ b/tensorflow/python/distribute/cluster_resolver/tpu_cluster_resolver.py @@ -109,11 +109,11 @@ def _tpu_service(self): if self._discovery_url: return discovery.build( - 'tpu', 'v1alpha1', credentials=credentials, + 'tpu', 'v1', credentials=credentials, discoveryServiceUrl=self._discovery_url, cache_discovery=False) else: return discovery.build( - 'tpu', 'v1alpha1', credentials=credentials, cache_discovery=False) + 'tpu', 'v1', credentials=credentials, cache_discovery=False) def _request_compute_metadata(self, path): req = Request('%s/computeMetadata/v1/%s' % (_GCE_METADATA_ENDPOINT, path),