fix: low recall if the num partitions is more than num rows

fix: low recall if the num partitions is more than num rows #1189