yu-val-weiss
commited on
Commit
·
a637547
1
Parent(s):
04c0ccd
Update blimp.py
Browse files
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 |
-
|
198 |
-
|
199 |
-
|
200 |
-
|
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 |
|