Allow mnist data to be writen to remote file system. (tensorflow#3890)

qlzh727 · web-flow · commit 2b4f7b443e20 · 2018-04-06T10:26:33.000-07:00
diff --git a/official/mnist/dataset.py b/official/mnist/dataset.py
@@ -70,7 +70,8 @@ def download(directory, filename):
   zipped_filepath = filepath + '.gz'
   print('Downloading %s to %s' % (url, zipped_filepath))
   urllib.request.urlretrieve(url, zipped_filepath)
-  with gzip.open(zipped_filepath, 'rb') as f_in, open(filepath, 'wb') as f_out:
+  with gzip.open(zipped_filepath, 'rb') as f_in, \
+      tf.gfile.Open(filepath, 'wb') as f_out:
     shutil.copyfileobj(f_in, f_out)
   os.remove(zipped_filepath)
   return filepath
