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