for i, (qnum, action) in enumerate(zip(batch.qids, actions)):
if isinstance(qnum, np.ndarray):
qnum = qnum.tolist()
finals[qnum] = np.argmax(ys[i][-1][:N_GUESSERS]).tolist()
buzzes[qnum] = [-1, -1]
for pos, chosen in enumerate(action):
if chosen < N_GUESSERS:
buzzes[qnum] = (pos, chosen)
After Change
if isinstance(qnum, np.ndarray):
qnum = qnum.tolist()
total = int(sum(mask))
buzzes[qnum] = scores[:total].tolist()
progress_bar(*test_iter.epoch_detail)
test_iter.finalize(reset=True)
progress_bar.finalize()
return buzzes
