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 | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/run_config.py b/tensorflow/contrib/learn/python/learn/estimators/run_config.py index 25da51675a..109c8d25e1 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/run_config.py +++ b/tensorflow/contrib/learn/python/learn/estimators/run_config.py @@ -89,9 +89,9 @@ class ClusterConfig(object): ``` cluster = {'ps': ['host1:2222', 'host2:2222'], 'worker': ['host3:2222', 'host4:2222', 'host5:2222']} - os.environ['TF_CONFIG'] = json.dumps({ + os.environ['TF_CONFIG'] = json.dumps( {'cluster': cluster, - 'task': {'type': 'worker', 'index': 1}}}) + 'task': {'type': 'worker', 'index': 1}}) config = ClusterConfig() assert config.master == 'host4:2222' assert config.task_id == 1 |