diff options
| author |  Brennan Saeta <saeta@google.com> | 2018-08-02 12:42:39 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-02 12:46:28 -0700 |
| commit | debcc45d2dca24a6914fc823477e5a1a43be3028 (patch) | |
| tree | 85bde1a6bae3f6e78e88b05b5ebc0d567a700ed0 /tensorflow/contrib/cluster_resolver | |
| parent | 97881c2675b2a746852011861c83564483eb4835 (diff) | |
[tpu]: Have a more helpful error message when no TPU name is specified.
PiperOrigin-RevId: 207147507
Diffstat (limited to 'tensorflow/contrib/cluster_resolver')
| -rw-r--r-- | tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py | 3 |
1 files changed, 3 insertions, 0 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 f9dc3effd0..1ab150d74a 100644 --- a/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py +++ b/tensorflow/contrib/cluster_resolver/python/training/tpu_cluster_resolver.py @@ -148,6 +148,9 @@ class TPUClusterResolver(ClusterResolver): else: tpu = self._envVarFallback() + if tpu is None: + raise ValueError('Please provide a TPU Name to connect to.') + self._tpu = compat.as_bytes(tpu) # self._tpu is always bytes self._job_name = job_name self._credentials = credentials |