Skip to content

Commit 0455a91

Browse files
committed
Fix nltk errors
1 parent 2756c6f commit 0455a91

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

benchmarks/eval_accuracy.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,10 @@ def eval_accuracy(request_outputs_dict, match_type):
131131

132132
else:
133133
metric = evaluate.load("rouge")
134-
nltk.download("punkt_tab")
134+
nltk.download("punkt_tab")
135+
nltk.download("punkt")
136+
nltk.download('wordnet')
137+
nltk.download('omw-1.4')
135138
preds, targets = postprocess_text(preds, targets)
136139
result = metric.compute(
137140
predictions=preds,

0 commit comments

Comments
 (0)