# pad out to N items if we're returned fewer
missing_items = N - len(batch_ids)
if missing_items > 0:
batch_ids = np.append(batch_ids, np.full(missing_items, -1))
batch_scores = np.append(
batch_scores, np.full(missing_items, -np.finfo(np.float32).max)
)