diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/run_config.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/run_config.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/run_config.py b/tensorflow/contrib/learn/python/learn/estimators/run_config.py index c5f8c110e6..c07736e701 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/run_config.py +++ b/tensorflow/contrib/learn/python/learn/estimators/run_config.py @@ -84,7 +84,7 @@ class ClusterConfig(object): 'worker': ['host3:2222', 'host4:2222', 'host5:2222']} os.environ['TF_CONFIG'] = json.dumps({ {'cluster': cluster, - 'task_id': {'type': 'worker', 'index': 1}}}) + 'task': {'type': 'worker', 'index': 1}}}) config = ClusterConfig() assert config.master == 'host4:2222' assert config.task_id == 1 |