This commit is contained in:
Nick
2019-09-26 16:39:51 -06:00
parent 0657dc6960
commit 1cd4d0c6f4
2 changed files with 4 additions and 4 deletions
+1 -1
View File
@@ -217,7 +217,7 @@ class CTRLGenerator():
# disallow some tokens
forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
"Edit", "&@@", "2:","1:", ":"]
"Edit", "&@@", "2:","1:", ":", "Edit@@"]
if num_new_lines > self.max_new_lines:
forbidden_tokens.append("\n")
+3 -3
View File
@@ -39,9 +39,9 @@ class Story():
self.results.append(story_block)
def latest_result(self):
# if len(self.results) > 1:
# return self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
if len(self.results) >= 1:
if len(self.results) > 1:
return self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
elif len(self.results) >= 1:
return self.story_start + self.actions[-1] + self.results[-1]
else:
return self.story_start