Test dataset length, item format, batch structure, prediction mode, and loader behavior to lock down the new data API.
Test dataset length, item format, batch structure, prediction mode, and loader behavior to lock down the new data API.