Skip to content

Commit 64ecec9

Browse files
committed
Add None check
1 parent a25e2e4 commit 64ecec9

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

machine/translation/huggingface/hugging_face_nmt_engine.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,9 +164,8 @@ 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(
168-
exp(output["sequence_score"]) if output["sequence_score"] is not None else -1
169-
)
167+
if output["sequence_score"] is not None:
168+
builder.set_sequence_confidence(exp(output["sequence_score"]))
170169
word_pairs: Optional[Collection[Union[AlignedWordPair, Tuple[int, int]]]] = None
171170
if output.get("token_attentions") is not None:
172171
src_indices = torch.argmax(output["token_attentions"], dim=1).tolist()

0 commit comments

Comments
 (0)