Skip to content

Commit 98dbe42

Browse files
committed
Fix domain field can be None
1 parent 9a28bb7 commit 98dbe42

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

sacrebleu/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import itertools
12
import json
23
import os
34
import re
@@ -548,7 +549,8 @@ def filter_subset(systems, test_sets, langpair, origlang, subset=None):
548549
from .dataset.wmt_xml import WMTXMLDataset
549550
if isinstance(dataset, WMTXMLDataset):
550551
fields = dataset._unwrap_wmt21_or_later(rawfile)
551-
for doc_origlang, doc_domain in zip(fields['origlang'], fields['domain']):
552+
domains = fields['domain'] if 'domain' in fields else itertools.repeat(None)
553+
for doc_origlang, doc_domain in zip(fields['origlang'], domains):
552554
if origlang is None:
553555
include_doc = True
554556
else:

0 commit comments

Comments
 (0)