aboutsummaryrefslogtreecommitdiff
path: root/markov/markov.py
diff options
context:
space:
mode:
Diffstat (limited to 'markov/markov.py')
-rw-r--r--markov/markov.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/markov/markov.py b/markov/markov.py
index a61a3ce..2a37a6c 100644
--- a/markov/markov.py
+++ b/markov/markov.py
@@ -365,6 +365,7 @@ class Markov(commands.Cog):
)
completion_count = row[0]
+ next_token = None
for i in range(MAX_TOKEN_GENERATION_ITERATIONS):
row = await (
await db.execute(
@@ -405,7 +406,10 @@ class Markov(commands.Cog):
" all of them. This should never happen!"
)
else:
- token = ""
+ # If we went through MAX_TOKEN_GENERATION_ITERATIONS completions
+ # without selecting any, then just select the last one we considered
+ # (round off the probability, effectively).
+ token = next_token
if token == "":
break