Mercurial > pylearn
diff pylearn/datasets/config.py @ 653:d3d8f5a17909
print warning on undefined PYLEARN_DATA_ROOT
author | bergstra@mlp4.ais.sandbox |
---|---|
date | Wed, 11 Feb 2009 01:42:58 -0500 |
parents | b054271b2504 |
children | 14d22ca1c8b5 |
line wrap: on
line diff
--- 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')