Merge pull request #191 from tqchen/master

[EXAMPLE] make example run on any folder
apache · Sep 30, 2015 · 6155880 · 6155880
2 parents 8b9e1c6 + 2df2ce7
commit 6155880
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/example/cifar10/cifar10.py b/example/cifar10/cifar10.py
@@ -1,10 +1,13 @@
 # pylint: skip-file
 import sys, os
-import mxnet as mx
-sys.path.append("../../tests/python/common")
+# code to automatically download dataset
+curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
+sys.path.append(os.path.join(curr_path, "../../tests/python/common"))
 import get_data
+import mxnet as mx
 import numpy as np
 import logging
+
 logger = logging.getLogger()
 logger.setLevel(logging.DEBUG)
 

diff --git a/example/mnist/data.py b/example/mnist/data.py
@@ -1,8 +1,10 @@
 # pylint: skip-file
 """ data iterator for mnist """
 import sys
-sys.path.insert(0, "../../python/")
-sys.path.append("../../tests/python/common")
+import os
+# code to automatically download dataset
+curr_path = os.path.dirname(os.path.abspath(os.path.expanduser(__file__)))
+sys.path.append(os.path.join(curr_path, "../../tests/python/common"))
 import get_data
 import mxnet as mx