Skip to content

Commit 1fd9ceb

Browse files
committed
Fix debug helper function
1 parent 8cbf578 commit 1fd9ceb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

russian_text_stresser/debug_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ class StressOptions:
8181

8282
def print_stressed_text_with_grammar_analysis(text: str):
8383
"""Prints a sentence with stress and grammar analysis."""
84-
rts = RussianTextStresser(use_large_model=True)
84+
rts = RussianTextStresser(use_large_model=False)
8585
stressed_text = rts.stress_text(text)
8686
original_doc = rts._nlp(text)
8787
stressed_doc = rts._nlp(stressed_text)
@@ -90,7 +90,7 @@ def print_stressed_text_with_grammar_analysis(text: str):
9090
for token, stressed_token in zip(original_doc, stressed_doc):
9191
stress_options = rts.rd._cur.execute("""
9292
SELECT w.word_id, w.pos, w.canonical_form, fow.form_of_word_id, ct.tag_text FROM word w
93-
JOIN form_of_word fow ON w.word_id = fow.word_id
93+
LEFT JOIN form_of_word fow ON w.word_id = fow.word_id
9494
JOIN case_tags ct ON ct.form_of_word_id = fow.form_of_word_id
9595
WHERE w.word_lower_and_without_yo = ?
9696
""", (token.text.lower(),)).fetchall()

0 commit comments

Comments
 (0)