ef00c480c4122f107f7d1ad481991d38b8da4f20,test/unittest_agent.py,UnittestAgent,test_pretrain,#UnittestAgent#,64

Before Change

        agent.initialize()

        states_batch = list()
        actions_batch = list()
        terminal_batch = list()
        reward_batch = list()

        states = environment.reset()

After Change

        environment.close()

        for filename in os.listdir(path=self.__class__.directory):
            os.remove(path=os.path.join(self.__class__.directory, filename))
            assert filename.startswith("trace-")
        os.rmdir(path=self.__class__.directory)

        self.finished_test()

Instances