Closes tensorflow#32

PiperOrigin-RevId: 239054994
Kritikalcoder · Mar 18, 2019 · a1e9ca2 · a1e9ca2
1 parent f58891f
commit a1e9ca2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/research/pate_2017/input.py b/research/pate_2017/input.py
@@ -250,7 +250,7 @@ def extract_mnist_data(filename, num_images, image_size, pixel_depth):
       np.save(filename, data)
       return data
   else:
-    with tf.gfile.Open(filename+'.npy', mode='r') as file_obj:
+    with tf.gfile.Open(filename+'.npy', mode='rb') as file_obj:
       return np.load(file_obj)
 
 
@@ -266,7 +266,7 @@ def extract_mnist_labels(filename, num_images):
       np.save(filename, labels)
     return labels
   else:
-    with tf.gfile.Open(filename+'.npy', mode='r') as file_obj:
+    with tf.gfile.Open(filename+'.npy', mode='rb') as file_obj:
       return np.load(file_obj)