Remove optional import

Vuizur · Aug 7, 2023 · c1e7d05 · c1e7d05
1 parent 681707a
commit c1e7d05
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/russian_text_stresser/text_stresser.py b/russian_text_stresser/text_stresser.py
@@ -5,7 +5,6 @@
  has_acute_accent_or_only_one_syllable,
 )
 from russian_text_stresser.russian_dictionary import RussianDictionary
-from russian_text_stresser.benchmark_word_sense_disambiguation import LocalLLM
 from russian_text_stresser.gpt3_WSD import WordSenseDisambiguator
 
 
@@ -15,7 +14,7 @@ def __init__(
  db_file: str = "russian_dict.db",
  simple_cases_file: Optional[str] = "simple_cases.pkl",
  use_large_model=False,
- llm: Optional[LocalLLM] = None, #
+ llm: Optional["LocalLLM"] = None, #
  add_stressrnn: bool = False,
  stressrnn_threshold: float = 0.8,
  ) -> None: