IDSIA / brainstorm

Fast, flexible and fun neural networks.
Other
1.3k stars 154 forks source link

Bugfix for brainstorm.tools.extract #108

Open michaelwand opened 8 years ago

michaelwand commented 8 years ago

Bugfix: brainstorm.tools.extract crashed when a mask was provided by the data iterator.

I also reimplemented extract_and_save in terms of extract (fixes the very same bug). But this glosses over some of the original settings in extract_and_save (e.g. chunked storage), which I believe are unimportant. Still, you may want to have a look.

I also added a test (test_tools.py) for the bug and its fix.