MAINT: Change shape of supplemental data to match batch data.

Expect the same shape of data for the supplemental data, to make
working and preparing with the supplemental data consistent with
what is passed to the algorithm.