togokah commited on
Commit
4e57705
1 Parent(s): b36f3a9

Update esp_sentence_bert.py

Browse files
utilities_language_bert/esp_sentence_bert.py CHANGED
@@ -213,11 +213,10 @@ class TASK:
213
 
214
  def inflect_distractors(self):
215
  inflected_distractors = []
 
 
 
216
  for distractor_lemma in self.distractors:
217
- if distractor_lemma is None:
218
- self.bad_target_word = True
219
- self.inflected_distractors = None
220
- continue
221
  if distractor_lemma.count('_') > 1:
222
  if distractor_lemma.startswith('haber_'):
223
  distractor_lemma = distractor_lemma.split('_')[-2]
 
213
 
214
  def inflect_distractors(self):
215
  inflected_distractors = []
216
+ if self.distractors is None:
217
+ self.bad_target_word = True
218
+ return
219
  for distractor_lemma in self.distractors:
 
 
 
 
220
  if distractor_lemma.count('_') > 1:
221
  if distractor_lemma.startswith('haber_'):
222
  distractor_lemma = distractor_lemma.split('_')[-2]