This commit is contained in:
Nick
2019-10-29 14:11:29 -06:00
parent 2c47e32d9f
commit 2f2a740685
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -47,7 +47,7 @@ def play_constrained():
generator = CTRLGenerator()
prompt = get_story_start("apocalypse")
context = get_context("apocalypse")
story_manager = ConstrainedStoryManager(generator)
story_manager = CTRLStoryManager(generator)
story_manager.start_new_story(prompt, context=context)
possible_actions = story_manager.get_possible_actions()
+1 -1
View File
@@ -210,7 +210,7 @@ class CTRLStoryManager(ConstrainedStoryManager):
def get_action_results_generate(self):
results = []
options = {}
options = {"word_blacklist":[]}
for phrase in self.action_phrases:
result = self.generate_action_result(self.story_context(), phrase, options=options)
action_verb = result[0].split()