Fix for "exclude" functionality
Summary
This MR introduces the following changes
- the exclude functionality doesn't work right now and causes a mismatch between expected dataframe shape and actual dataframe shape:
TypeError: `generator` yielded an element of shape (15000, 24) where an element of shape (None, 40) was expected.
Relates to the following issues
- Small bugfix related to the issue 156