Fix import that was removed by commit 7bd463dabd7d5d53b7097a3e00eaac79e9...

...dfff05

This isn't released yet and caused buidlbot failure i Pylearn (not pylearn2)