When requesting 0 demos (train or valid), 1 demo was being generated and saved. This commit fixes this.
When requesting 0 demos (train or valid), 1 demo was being generated and saved. This commit fixes this.

mila-iqia/babyai