Update module example (apache#12961)

* Update Module example * trigger CI
jlcontreras · Nov 13, 2018 · d24783b · d24783b
1 parent e7bd066
commit d24783b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/example/utils/__init__.py → example/module/utils/__init__.py b/example/utils/__init__.py → example/module/utils/__init__.py
diff --git a/example/utils/get_data.py → example/module/utils/get_data.py b/example/utils/get_data.py → example/module/utils/get_data.py
@@ -17,6 +17,7 @@
 
 import os
 import mxnet as mx
+import zipfile
 
 def get_mnist(data_dir):
     if not os.path.isdir(data_dir):
@@ -28,7 +29,7 @@ def get_mnist(data_dir):
        (not os.path.exists('t10k-labels-idx1-ubyte')):
         import urllib, zipfile
         zippath = os.path.join(os.getcwd(), "mnist.zip")
-        urllib.urlretrieve("http://data.mxnet.io/mxnet/data/mnist.zip", zippath)
+        mx.test_utils.download("http://data.mxnet.io/mxnet/data/mnist.zip", zippath)
         zf = zipfile.ZipFile(zippath, "r")
         zf.extractall()
         zf.close()
@@ -45,7 +46,7 @@ def get_cifar10(data_dir):
         import urllib, zipfile, glob
         dirname = os.getcwd()
         zippath = os.path.join(dirname, "cifar10.zip")
-        urllib.urlretrieve("http://data.mxnet.io/mxnet/data/cifar10.zip", zippath)
+        mx.test_utils.download("http://data.mxnet.io/mxnet/data/cifar10.zip", zippath)
         zf = zipfile.ZipFile(zippath, "r")
         zf.extractall()
         zf.close()
@@ -56,7 +57,6 @@ def get_cifar10(data_dir):
         os.rmdir(os.path.join(dirname, "cifar"))
     os.chdir(cwd)
 
-# data
 def get_cifar10_iterator(args, kv):
     data_shape = (3, 28, 28)
     data_dir = args.data_dir