Upd bert version

utilities_language_bert/esp_sentence_bert.py

@@ -232,12 +232,14 @@ class TASK:
         if len(inflected_distractors) < num_distractors:
             self.bad_target_word = True
         else:
+            self.distractors_number = num_distractors
             self.inflected_distractors = inflected_distractors
 
     def sample_distractors(self, num_distractors):
         if not self.bad_target_word:
             num_distractors = min(self.distractors_number, num_distractors) if num_distractors >= 4 else num_distractors
-            self.distractors = sample(self.distractors[:min(self.distractors_number, 10)], num_distractors)
+            self.inflected_distractors = sample(self.inflected_distractors[:min(self.distractors_number, 10)],
+                                                num_distractors)
 
     def compile_task(self, max_num_distractors):
         len_distractors = len(self.inflected_distractors)

utilities_language_w2v/esp_main_workflow_w2v.py

@@ -30,7 +30,6 @@ def main_workflow(
         target_words: str,
         num_distractors: int,
         save_name: str,
-        distractor_model,
         global_bad_target_words=BAD_USER_TARGET_WORDS):
     """
     This is the main course of the program.
@@ -50,7 +49,6 @@ def main_workflow(
     :param num_distractors: how many distractors does the user want the task to contain
     :param save_name: user specifies name to save file in cloud
     :param global_bad_target_words:global_bad_target_words
-    :param distractor_model which semantic model is used to find distractors
     :return: Dictionary with output data: filename, amount_mode, text_with_gaps, tasks_as_list, correct_answers,
              student_out, teacher_out, total_out, original_text
     """