31803d30a9f0c4dde1342678f8fced1d0978c2db,tests/test_torch_agent.py,TestTorchAgent,test_batchify,#TestTorchAgent#,306

Before Change

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

            // test lambda
            batch = agent.batchify(obs_vecs, is_valid=(
                lambda obs: "text_vec" in obs and len(obs["text_vec"]) < 3))
            self.assertEqual(batch.text_vec.tolist(), [[1, 2]])
            self.assertEqual(batch.text_lengths, [2])
            self.assertEqual(batch.label_vec.tolist(), [[1, 2]])
            self.assertEqual(batch.label_lengths, [2])

After Change

                return "text_vec" in obs and len(obs["text_vec"]) < 3
            agent.is_valid = is_valid

            batch = agent.batchify(obs_vecs)
            self.assertEqual(batch.text_vec.tolist(), [[1, 2]])
            self.assertEqual(batch.text_lengths, [2])
            self.assertEqual(batch.label_vec.tolist(), [[1, 2]])
            self.assertEqual(batch.label_lengths, [2])

Instances