I think you forgot to import pylearn2 here