9d22742baeeb608bac5594c3ece96f62f0734dce,qanta/buzzer/trainer.py,Trainer,test,#Trainer#Any#,62

Before Change

            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

Instances