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
