diff options
author | 2017-11-22 15:51:33 -0800 | |
---|---|---|
committer | 2017-11-22 15:55:29 -0800 | |
commit | 8d9eda26be345ace2e110feb0cf9a2500990eb82 (patch) | |
tree | b23811daa852c647a3cd6312b5005ee30e51b4f9 /tensorflow/contrib/cluster_resolver | |
parent | c0b8a07776a7f64fc45113159230a0a10273ec4a (diff) |
Remove hardcoded discovery document now that the TPU alpha API definitions are public (https://www.googleapis.com/discovery/v1/apis/tpu/v1alpha1/rest).
PiperOrigin-RevId: 176710985
Diffstat (limited to 'tensorflow/contrib/cluster_resolver')
-rw-r--r-- | tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py index f0144e9faa..c74da9cabd 100644 --- a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py +++ b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py @@ -80,13 +80,9 @@ class TPUClusterResolver(ClusterResolver): raise ImportError('googleapiclient must be installed before using the ' 'TPU cluster resolver') - # TODO(b/67375680): Remove custom URL once TPU APIs are finalized self._service = discovery.build( - 'tpu', - 'v1', - credentials=self._credentials, - discoveryServiceUrl='https://storage.googleapis.com' - '/tpu-api-definition/v1alpha1.json') + 'tpu', 'v1alpha1', + credentials=self._credentials) else: self._service = service |