# HG changeset patch # User bergstra@mlp4.ais.sandbox # Date 1234334578 18000 # Node ID d3d8f5a17909b5dfb7698d50e09ad6c2906bc8c7 # Parent 37a259d6946a4b7afbe63f010563f7fd8951b2c0 print warning on undefined PYLEARN_DATA_ROOT diff -r 37a259d6946a -r d3d8f5a17909 pylearn/datasets/MNIST.py --- a/pylearn/datasets/MNIST.py Wed Feb 11 01:42:26 2009 -0500 +++ b/pylearn/datasets/MNIST.py Wed Feb 11 01:42:58 2009 -0500 @@ -7,7 +7,7 @@ import numpy from ..io.amat import AMat -from .config import data_root +from .config import data_root # config from .dataset import Dataset def head(n=10, path=None): diff -r 37a259d6946a -r d3d8f5a17909 pylearn/datasets/config.py --- a/pylearn/datasets/config.py Wed Feb 11 01:42:26 2009 -0500 +++ b/pylearn/datasets/config.py Wed Feb 11 01:42:58 2009 -0500 @@ -4,10 +4,13 @@ Especially, the locations of data files. """ -import os +import os, sys def env_get(key, default): + if os.getenv(key) is None: + print >> sys.stderr, "WARNING: Environment variable", key, + print >> sys.stderr, "is not set. Using default of", default return default if os.getenv(key) is None else os.getenv(key) def data_root(): - return env_get('PYLEARN_DATA_ROOT', '/u/bergstrj/pub/data/') + return env_get('PYLEARN_DATA_ROOT', os.getenv('HOME')+'/data')