From de718ab434a6cc5e9367d3dbb608950ee2979ff6 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Wed, 25 Sep 2019 12:57:04 -0600 Subject: [PATCH] update --- story/story_manager.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/story/story_manager.py b/story/story_manager.py index 387499a..25be17c 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -178,11 +178,10 @@ class CTRLStoryManager(ConstrainedStoryManager): options["word_whitelist"][0] = get_possible_verbs(type="movement") options["word_whitelist"][1] = ["to"] options["word_whitelist"][2] = ["the"] - options["word_whitelist"][3] = ["hospital"] - options["word_whitelist"][4] = \ + options["word_whitelist"][3] = \ [room for room in possible_rooms if room is not self.story.game_state["current_room"]] - options["word_whitelist"][5] = ["and"] - options["word_whitelist"][6] = ["see"] + options["word_whitelist"][4] = ["and"] + options["word_whitelist"][5] = ["see"] return options @@ -208,8 +207,10 @@ class CTRLStoryManager(ConstrainedStoryManager): options = self.get_constrained_movement_options() for phrase in self.action_phrases: result = self.generate_action_result(self.story_context(), phrase, options=options) - location = result[0].split()[4] - options["word_whitelist"][4].remove(location) + location = result[0].split()[3] + print("result is ", result) + print("location is ", location) + options["word_whitelist"][3].remove(location) results.append(result) return results