diff options
| author | Rose Hogenson <rhogenson@posteo.net> | 2022-02-16 20:19:47 -0800 |
|---|---|---|
| committer | Rose Hogenson <rhogenson@posteo.net> | 2022-02-16 20:19:47 -0800 |
| commit | 50714995a9f986b9e01feb12e6c537be93392484 (patch) | |
| tree | 4ad61d469100a5b2df80a922c1d1f3174dd5c242 /gpt_adventure.py | |
| parent | Merge branch 'main' of gitlab.com:rhogenson/gpt-adventure (diff) | |
| download | gpt-adventure-50714995a9f986b9e01feb12e6c537be93392484.tar.zst | |
Fix some lint errors.
Diffstat (limited to 'gpt_adventure.py')
| -rwxr-xr-x | gpt_adventure.py | 3 |
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 |
