Skip to content

Commit

Permalink
update test for the new behaviour
Browse files Browse the repository at this point in the history
  • Loading branch information
pavel-esir committed Dec 4, 2024
1 parent af05de5 commit 2a0f521
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/python_tests/test_generate_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -687,7 +687,7 @@ def test_unicode_pybind_decoding_3():
model_id, path = 'katuni4ka/tiny-random-phi3', Path('tiny-random-phi3')
pipe = read_model((model_id, path))[4]
res_str = []
pipe.generate(",", max_new_tokens=4, streamer=lambda x: res_str.append(x))
pipe.generate(",", max_new_tokens=4, streamer=lambda x: res_str.extend(list(x)))
assert '�' == res_str[-1]


Expand Down

0 comments on commit 2a0f521

Please sign in to comment.