Allow to import rnnrbm.py even with pylab is not available.

nouiz · nouiz · commit 5c0f472abe8b · 2013-02-13T09:58:54.000-05:00
This is needed for tests.
diff --git a/code/rnnrbm.py b/code/rnnrbm.py
@@ -4,10 +4,15 @@
 # More information at http://deeplearning.net/tutorial/rnnrbm.html
 
 import glob
-import numpy
-import pylab
 import sys
 
+import numpy
+try:
+    import pylab
+except ImportError:
+    print "pylab isn't available, if you use their fonctionality, it will crash"
+    print "It can be installed with 'pip install -q Pillow'"
+
 from midi.utils import midiread, midiwrite
 import theano
 import theano.tensor as T
