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 |
|