# HG changeset patch
# User gdesjardins
# Date 1309878111 14400
# Node ID 83d3c9ee6d65f8bc0e32109d41a4d975c8dad05b
# Parent  4727a7e4d506a858767e7b4a665d9d693bab1b4c
* changed MNIST dataset to use config.get_filepath_in_roots mechanism

diff -r 4727a7e4d506 -r 83d3c9ee6d65 pylearn/datasets/MNIST.py
--- a/pylearn/datasets/MNIST.py	Tue Jul 05 10:58:19 2011 -0400
+++ b/pylearn/datasets/MNIST.py	Tue Jul 05 11:01:51 2011 -0400
@@ -6,8 +6,8 @@
 import numpy
 
 from pylearn.io.pmat import PMat
-from pylearn.datasets.config import data_root # config
 from pylearn.datasets.dataset import Dataset
+import config
 
 def head(n=10, path=None):
     """Load the first MNIST examples.
@@ -18,7 +18,9 @@
 
     """
     if path is None:
-      path = os.path.join(data_root(), 'mnist','mnist_all.pmat')
+        # dataset lookup through $PYLEARN_DATA_ROOT
+        _path = os.path.join('mnist', 'mnist_all.pmat')
+        path = config.get_filepath_in_roots(_path)
 
     dat = PMat(fname=path)