From 50714995a9f986b9e01feb12e6c537be93392484 Mon Sep 17 00:00:00 2001 From: Rose Hogenson Date: Wed, 16 Feb 2022 20:19:47 -0800 Subject: Fix some lint errors. --- gpt_adventure.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'gpt_adventure.py') 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 -- cgit v1.3.1