yu-val-weiss commited on
Commit
a637547
·
1 Parent(s): 04c0ccd

Update blimp.py

Browse files
Files changed (1) hide show
  1. blimp.py +10 -5
blimp.py CHANGED
@@ -194,11 +194,16 @@ class Blimp(evaluate.Metric):
194
  tokenizer.add_special_tokens({"pad_token": existing_special_tokens[0]})
195
 
196
  run_all = len(predictions) == 0 or predictions[0] == "*"
197
- blimp_sets = (
198
- BLIMP_UIDS
199
- if run_all
200
- else [p for p in BLIMP_UIDS if p.lower() in predictions]
201
- )
 
 
 
 
 
202
 
203
  assert len(blimp_sets) > 0, "no valid phenomena selected"
204
 
 
194
  tokenizer.add_special_tokens({"pad_token": existing_special_tokens[0]})
195
 
196
  run_all = len(predictions) == 0 or predictions[0] == "*"
197
+
198
+ if run_all:
199
+ blimp_sets = BLIMP_UIDS
200
+ else:
201
+ blimp_sets = []
202
+ for p in predictions:
203
+ if p in BLIMP_UIDS:
204
+ blimp_sets.append(p)
205
+ else:
206
+ logging.logging.warning(f"{p} not a valid UID, skipping...")
207
 
208
  assert len(blimp_sets) > 0, "no valid phenomena selected"
209