Skip to content

Commit 13d1b1a

Browse files
committed
Do not exp() None
1 parent 797bf75 commit 13d1b1a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

machine/translation/huggingface/hugging_face_nmt_engine.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,9 @@ def _try_translate_n_batch(
164164
builder = TranslationResultBuilder(input_tokens)
165165
for token, score in zip(output["translation_tokens"], output["token_scores"]):
166166
builder.append_token(token, TranslationSources.NMT, exp(score))
167-
builder.set_sequence_confidence(exp(output["sequence_score"]))
167+
builder.set_sequence_confidence(
168+
exp(output["sequence_score"]) if output["sequence_score"] is not None else -1
169+
)
168170
word_pairs: Optional[Collection[Union[AlignedWordPair, Tuple[int, int]]]] = None
169171
if output.get("token_attentions") is not None:
170172
src_indices = torch.argmax(output["token_attentions"], dim=1).tolist()

0 commit comments

Comments
 (0)