a4f6690b9c4290a7f47761100523ff2ef5335964,tests/agents_tests/test_trpo.py,TestTRPO,_test_abc,#TestTRPO#Any#Any#Any#Any#,183

Before Change

        n_test_runs = 5
        for _ in range(n_test_runs):
            total_r = 0.0
            obs = test_env.reset()
            done = False
            reward = 0.0
            while not done:
                action = agent.act(obs)
                obs, reward, done, _ = test_env.step(action)
                total_r += reward
            agent.stop_episode()

            if require_success:
                self.assertAlmostEqual(total_r, successful_return)

After Change

        )
        if require_success:
            n_succeeded = np.sum(np.asarray(eval_returns) >= successful_return)
            self.assertEqual(n_succeeded, n_test_runs)

        // Save
        agent.save(self.agent_dirname)

Instances