Fix inflection in bert

pages/2_👨‍🏫_Начало_работы.py

@@ -81,7 +81,7 @@ if st.session_state.get('-LOGGED_IN_BOOL-'):
         with COL2:
             UPLOAD_CLOUD_CEFR_LEVEL = custom_select_box(
                 'Выберите языковой уровень',
-                ['A1', 'A2', 'B1', 'B2', 'C1', 'C2', 'Нет'],
+                ['A1', 'A2', 'B1', 'B2', 'C1', 'C2', 'Без уровня'],
                 no_selection_label='-Выберите языковой уровень-')
             st.session_state['-UPLOAD_CLOUD_CEFR_LEVEL-'] = UPLOAD_CLOUD_CEFR_LEVEL
         LOAD_BUTTON = LOAD_FORM.form_submit_button('Загрузить')

utilities_language_bert/esp_sentence_bert.py

@@ -220,11 +220,11 @@ class TASK:
             if distractor_lemma.count('_') > 1:
                 if distractor_lemma.startswith('haber_'):
                     distractor_lemma = distractor_lemma.split('_')[-2]
-                    inflected = inflect(lemma=distractor_lemma, target_pos=self.pos, target_tags=self.tags)
+                    inflected = inflect(lemma=distractor_lemma, target_pos=self.pos[1], target_tags=self.tags)
                 else:
                     continue
             else:
-                inflected = inflect(lemma=distractor_lemma, target_pos=self.pos, target_tags=self.tags)
+                inflected = inflect(lemma=distractor_lemma, target_pos=self.pos[1], target_tags=self.tags)
             if inflected is not None:
                 inflected_distractors.append(inflected)
         num_distractors = min(4, self.max_num_distractors) if self.max_num_distractors >= 4 \

utilities_language_general/esp_utils.py

@@ -184,12 +184,7 @@ def get_distractors_from_model_bert(model, text_with_masked_task: str, lemma: st
                                     max_num_distractors: int, max_length_ratio=5, min_edit_distance_ratio=0.5):
     _distractors = []
     try:
-        if distractor_minimum:
-            bert_candidates = [token for token in model(text_with_masked_task,
-                                                        top_k=max_num_distractors + 100,
-                                                        targets=list(distractor_minimum))]
-        else:
-            bert_candidates = [token for token in model(text_with_masked_task, top_k=max_num_distractors + 100)]
+        bert_candidates = [token for token in model(text_with_masked_task, top_k=max_num_distractors + 100)]
         candidates = []
         for candidate in bert_candidates:
             if isinstance(candidate, list):