Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

Commit e4fe66c

Browse files
author
Ryan Sepassi
committed
Tweak TF_CONFIG script and bump version to 1.0.9
PiperOrigin-RevId: 160563166
1 parent 22ca232 commit e4fe66c

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
setup(
77
name='tensor2tensor',
8-
version='1.0.8',
8+
version='1.0.9',
99
description='Tensor2Tensor',
1010
author='Google Inc.',
1111
author_email='[email protected]',

tensor2tensor/bin/make_tf_configs.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@
3232

3333
# Dependency imports
3434

35-
import six
3635
import tensorflow as tf
3736

3837
flags = tf.flags
@@ -51,7 +50,7 @@ def main(_):
5150

5251
cluster = {"ps": ps, "worker": workers}
5352

54-
for task_type, jobs in six.iteritems(cluster):
53+
for task_type, jobs in (("worker", workers), ("ps", ps)):
5554
for idx, job in enumerate(jobs):
5655
if task_type == "worker":
5756
cmd_line_flags = " ".join([
@@ -77,7 +76,7 @@ def main(_):
7776
"index": idx
7877
}
7978
})
80-
print(tf_config + "\t" + cmd_line_flags)
79+
print("'%s'\t%s" % (tf_config, cmd_line_flags))
8180

8281

8382
if __name__ == "__main__":

0 commit comments

Comments
 (0)