Skip to content

Commit

Permalink
fix black
Browse files Browse the repository at this point in the history
  • Loading branch information
minhthuc2502 committed Aug 13, 2024
1 parent cfab91f commit ddd3f80
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions python/tests/test_translator.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,9 +111,12 @@ def test_batch_translation(max_batch_size):
assert output[0].scores[0] < 0
assert not output[0].attention

expected_repr = "TranslationResult(hypotheses=%s, scores=%s, attention=[], log_probs=[])" % (
output[0].hypotheses,
output[0].scores,
expected_repr = (
"TranslationResult(hypotheses=%s, scores=%s, attention=[], log_probs=[])"
% (
output[0].hypotheses,
output[0].scores,
)
)
assert repr(output[0]) == expected_repr

Expand Down

0 comments on commit ddd3f80

Please sign in to comment.