Sort the commands

microsoft · Sep 27, 2024 · b6af81a · b6af81a
1 parent f22f086
commit b6af81a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/textworld/generator/game.py b/textworld/generator/game.py
@@ -596,16 +596,16 @@ def possible_admissible_commands(self) -> List[str]:
             type2names[f'{{{type}}}'].append(name)
 
         templates = [re.sub(r"{.*?}", "{{{}}}", template).format(*mappings)
-                    for template in self.command_templates
-                    for mappings in product(*[[arg.strip('{}')] + self.kb.types.descendants(arg.strip('{}'))
-                                              for arg in re.findall(r"{.*?}", template)])]
+                     for template in self.command_templates
+                     for mappings in product(*[[arg.strip('{}')] + self.kb.types.descendants(arg.strip('{}'))
+                                               for arg in re.findall(r"{.*?}", template)])]
 
         templates = sorted(set(templates))
-
         commands = [re.sub(r"{.*?}", "{}", template).format(*mappings)
                     for template in templates
                     for mappings in product(*[type2names[arg] for arg in re.findall(r"{.*?}", template)])]
-        return commands
+
+        return sorted(commands)
 
     @property
     def objective(self) -> str: