comparison 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
comparison
equal deleted inserted replaced
652:37a259d6946a 653:d3d8f5a17909
2 2
3 3
4 Especially, the locations of data files. 4 Especially, the locations of data files.
5 """ 5 """
6 6
7 import os 7 import os, sys
8 def env_get(key, default): 8 def env_get(key, default):
9 if os.getenv(key) is None:
10 print >> sys.stderr, "WARNING: Environment variable", key,
11 print >> sys.stderr, "is not set. Using default of", default
9 return default if os.getenv(key) is None else os.getenv(key) 12 return default if os.getenv(key) is None else os.getenv(key)
10 13
11 def data_root(): 14 def data_root():
12 return env_get('PYLEARN_DATA_ROOT', '/u/bergstrj/pub/data/') 15 return env_get('PYLEARN_DATA_ROOT', os.getenv('HOME')+'/data')
13 16