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')