summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorRose Hogenson <rhogenson@posteo.net>2022-02-16 20:19:47 -0800
committerRose Hogenson <rhogenson@posteo.net>2022-02-16 20:19:47 -0800
commit50714995a9f986b9e01feb12e6c537be93392484 (patch)
tree4ad61d469100a5b2df80a922c1d1f3174dd5c242
parent549db4ef1922ef40569ee89eb87c256d7b96b6af (diff)
downloadgpt-adventure-50714995a9f986b9e01feb12e6c537be93392484.tar.zst
Fix some lint errors.
-rwxr-xr-xgpt_adventure.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/gpt_adventure.py b/gpt_adventure.py
index c2137db..6fda9fe 100755
--- a/gpt_adventure.py
+++ b/gpt_adventure.py
@@ -3,7 +3,6 @@
"""GPT adventure is a text-adventure style game powered by AI."""
import argparse
-import re
import shutil
import sys
@@ -13,6 +12,7 @@ import transformers
def load_nltk():
+ """Download required nltk modules."""
try:
tokenize.sent_tokenize("")
except LookupError:
@@ -20,6 +20,7 @@ def load_nltk():
def complete_sentence(snippet: str) -> bool:
+ """Return whether the snippet ends in a complete sentence."""
sentences = tokenize.sent_tokenize(snippet)
return len(tokenize.sent_tokenize(sentences[-1] + " extra")) == 2