Skip to content

Commit c313c28

Browse files
fix backcompatibility in translator
1 parent 3f80e13 commit c313c28

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onmt/translate/translator.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -316,7 +316,8 @@ def translate(
316316
[('src', src_data), ('tgt', tgt_data)])
317317

318318
# corpus_id field is useless here
319-
self.fields.pop('corpus_id')
319+
if self.fields.get("corpus_id", None) is not None:
320+
self.fields.pop('corpus_id')
320321
data = inputters.Dataset(
321322
self.fields, readers=_readers, data=_data, dirs=_dir,
322323
sort_key=inputters.str2sortkey[self.data_type],

0 commit comments

Comments
 (0)