aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/test
diff options
context:
space:
mode:
authorGravatar Rohan Jain <rohanj@google.com>2017-01-25 13:42:13 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-01-25 14:03:58 -0800
commit39eca117a335918e5314f3593022568a0454ba31 (patch)
treee65b43a6246c9f5e2562e0abbf45748ed6a778de /tensorflow/tools/test
parent191658d54f90ac03c15b339326129cd52d1f56a3 (diff)
Adding support for 'b' in mode for FileIO.
Change: 145590579
Diffstat (limited to 'tensorflow/tools/test')
-rw-r--r--tensorflow/tools/test/system_info_lib.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/test/system_info_lib.py b/tensorflow/tools/test/system_info_lib.py
index 69d71b3a6f..40e520106e 100644
--- a/tensorflow/tools/test/system_info_lib.py
+++ b/tensorflow/tools/test/system_info_lib.py
@@ -79,7 +79,7 @@ def gather_cpu_info():
# Gather num_cores_allowed
try:
- with gfile.GFile('/proc/self/status') as fh:
+ with gfile.GFile('/proc/self/status', 'rb') as fh:
nc = re.search(r'(?m)^Cpus_allowed:\s*(.*)$', fh.read())
if nc: # e.g. 'ff' => 8, 'fff' => 12
cpu_info.num_cores_allowed = (