Use constants' type as their name. (#293)

microsoft · Feb 3, 2022 · b94f39a · b94f39a
1 parent c419bb6
commit b94f39a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/textworld/generator/text_generation.py b/textworld/generator/text_generation.py
@@ -126,6 +126,7 @@ def generate_text_from_grammar(game, grammar: Grammar):
     # Assign name and description to objects.
     for obj in game.world.objects:
         if obj.type in ["I", "P"]:
+            obj.name = obj.type
             continue
 
         assign_name_to_object(obj, grammar, game.infos)