def test_load_textworld_games(): request_infos = EnvInfos() request_infos.admissible_commands = True request_infos.description = False request_infos.location = False request_infos.facts = False request_infos.last_action = False request_infos.game = True max_episode_steps = 100 batch_size = 5 name = "test" base_dir = "test-data/rl_games/" env = load_textworld_games( [ f"{base_dir}tw-cooking-recipe1+take1+cook+open-BNVaijeLTn3jcvneFBY2.z8", f"{base_dir}tw-cooking-recipe1+take1+open-BNVaijeLTn3jcvneFBY2.z8", f"{base_dir}tw-cooking-recipe1+take1+cook+open-BNVaijeLTn3jcvneFBY2.z8", f"{base_dir}tw-cooking-recipe1+take1+open-BNVaijeLTn3jcvneFBY2.z8", ], name, request_infos, max_episode_steps, batch_size, ) assert len(env.gamefiles) == 4 assert env.request_infos == request_infos assert env.batch_size == batch_size # for some reason env.spec.max_episode_steps is None # assert env.spec.max_episode_steps == max_episode_steps assert env.spec.id.split("-")[1] == name
def request_infos_for_eval() -> EnvInfos: request_infos = EnvInfos() request_infos.admissible_commands = True request_infos.description = True request_infos.location = True request_infos.facts = True request_infos.last_action = True request_infos.game = True return request_infos