1ba2dd41729cb7254e9e40b036e3d516a05d7a28,tests/test_torch_agent.py,TestTorchAgent,test_batchify,#TestTorchAgent#,310

Before Change

            self.assertEqual(list(batch.valid_indices), [1, 0, 2])

            // now sort just on ys
            new_vecs = [vecs.copy() for vecs in obs_vecs]
            for vec in new_vecs:
                vec.pop("text")
                vec.pop("text_vec")

After Change

            self.assertEqual(batch.labels, obs_batch[2][lab_key])
            self.assertEqual(list(batch.valid_indices), [2])

        agent.history.clear()
        obs_cands = [
            agent.vectorize({"label_candidates": ["A", "B", "C"]},
                            agent.history),
            agent.vectorize({"label_candidates": ["1", "2", "5", "3", "Sir"]},

Instances