huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 15:52:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test utils

Browse Source
This commit is contained in:
OlivierDehaene 2022-12-12 17:46:39 +01:00
parent 3eaf7ee2e3
commit 42ae103559
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
server/tests/test_utils.py
View File

@ -39,18 +39,18 @@ def test_stop_sequence_criteria_empty():
def test_stopping_criteria():
criteria = StoppingCriteria([StopSequenceCriteria([1, 2, 3])], max_new_tokens=5)
assert not criteria([1])
assert not criteria([1, 2])
assert criteria([1, 2, 3])
assert criteria([1]) == (False, None)
assert criteria([1, 2]) == (False, None)
assert criteria([1, 2, 3]) == (True, "stop_sequence")
def test_stopping_criteria_max():
criteria = StoppingCriteria([StopSequenceCriteria([1, 2, 3])], max_new_tokens=5)
assert not criteria([1])
assert not criteria([1, 1])
assert not criteria([1, 1, 1])
assert not criteria([1, 1, 1, 1])
assert criteria([1, 1, 1, 1, 1])
assert criteria([1]) == (False, None)
assert criteria([1, 1]) == (False, None)
assert criteria([1, 1, 1]) == (False, None)
assert criteria([1, 1, 1, 1]) == (False, None)
assert criteria([1, 1, 1, 1, 1]) == (True, "length")
def test_weight_hub_files():