gender check for some lvl

utilities_language_general/esp_utils.py

@@ -134,7 +134,7 @@ def get_distractors_from_model(model, lemma: str, pos: str, gender: str or None,
                               and distractor_pos in COMBINE_POS['simple'][level_name][pos] and pos in COMBINE_POS['simple'][level_name][distractor_pos])
                           )
                          and distractor_lemma != lemma
-                         and candidate_gender == gender
+                         and (candidate_gender == gender and level_name in ('B1', 'B2', 'C1', 'C2'))
                          and length_ratio <= max_length_ratio
                          and distractor_lemma not in global_distractors
                          and edit_distance(lemma, distractor_lemma) / ((len(lemma) + len(distractor_lemma)) / 2) >
@@ -210,7 +210,7 @@ def get_distractors_from_model_bert(model, text_with_masked_task: str, lemma: st
                  and distractor_pos in COMBINE_POS['simple'][level_name][pos] and pos in COMBINE_POS['simple'][level_name][distractor_pos]))
                 and distractor_lemma != lemma
                 and (len(_distractors) < max_num_distractors+100)
-                and (candidate_gender == gender)
+                and (candidate_gender == gender and level_name in ('B1', 'B2', 'C1', 'C2'))
                 and (length_ratio <= max_length_ratio)  # May be changed if case of phrases
                 and (distractor_lemma not in global_distractors)
                 and (edit_distance(lemma, distractor_lemma)  # May be changed if case of phrases