diff --git a/textworld/challenges/coin_collector.py b/textworld/challenges/coin_collector.py index e8a752b0..8e14e60b 100644 --- a/textworld/challenges/coin_collector.py +++ b/textworld/challenges/coin_collector.py @@ -89,7 +89,6 @@ def make_game(mode: str, options: GameOptions) -> textworld.Game: metadata["seeds"] = options.seeds metadata["world_size"] = options.nb_rooms metadata["quest_length"] = options.quest_length - metadata["grammar_flags"] = options.grammar.encode() rngs = options.rngs rng_map = rngs['map'] @@ -121,7 +120,7 @@ def make_game(mode: str, options: GameOptions) -> textworld.Game: # Add distractor rooms, if needed. chain_of_rooms = list(rooms) - while len(rooms) < n_rooms: + while len(rooms) < options.nb_rooms: if mode == "random": src = rng_map.choice(rooms) else: diff --git a/textworld/challenges/treasure_hunter.py b/textworld/challenges/treasure_hunter.py index 890efa9d..ea1dfe1e 100644 --- a/textworld/challenges/treasure_hunter.py +++ b/textworld/challenges/treasure_hunter.py @@ -118,7 +118,6 @@ def make_game(mode: str, options: GameOptions) -> textworld.Game: metadata["seeds"] = options.seeds metadata["world_size"] = options.nb_rooms metadata["quest_length"] = options.quest_length - metadata["grammar_flags"] = options.grammar.encode() rngs = options.rngs rng_map = rngs['seed_map']