mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-10 10:31:35 +00:00
7d29ddba42
* merge hf utils modules * relnotes * lint * Update releasenotes/notes/merge-hf-utils-modules-5c16e04025123568.yaml Co-authored-by: Stefano Fiorucci <stefanofiorucci@gmail.com> --------- Co-authored-by: Stefano Fiorucci <stefanofiorucci@gmail.com>
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import pytest
|
|
|
|
from haystack.utils.hf import check_generation_params
|
|
|
|
|
|
def test_empty_dictionary():
|
|
# no exception raised
|
|
check_generation_params({})
|
|
|
|
|
|
def test_valid_generation_parameters():
|
|
# these are valid parameters
|
|
kwargs = {"max_new_tokens": 100, "temperature": 0.8}
|
|
additional_accepted_params = None
|
|
check_generation_params(kwargs, additional_accepted_params)
|
|
|
|
|
|
def test_invalid_generation_parameters():
|
|
# these are invalid parameters
|
|
kwargs = {"invalid_param": "value"}
|
|
additional_accepted_params = None
|
|
with pytest.raises(ValueError):
|
|
check_generation_params(kwargs, additional_accepted_params)
|
|
|
|
|
|
def test_additional_accepted_params_empty_list():
|
|
kwargs = {"temperature": 0.8}
|
|
additional_accepted_params = []
|
|
check_generation_params(kwargs, additional_accepted_params)
|
|
|
|
|
|
def test_additional_accepted_params_known_parameter():
|
|
# both are valid parameters
|
|
kwargs = {"temperature": 0.8}
|
|
additional_accepted_params = ["max_new_tokens"]
|
|
check_generation_params(kwargs, additional_accepted_params)
|
|
|
|
|
|
def test_additional_accepted_params_unknown_parameter():
|
|
kwargs = {"strange_param": "value"}
|
|
additional_accepted_params = ["strange_param"]
|
|
# Although strange_param is not generation param the check_generation_params
|
|
# does not raise exception because strange_param is passed as additional_accepted_params
|
|
check_generation_params(kwargs, additional_accepted_params)
|