From b8e9984fe8e32ab06da54c0c8c181b141a7227f9 Mon Sep 17 00:00:00 2001 From: anakin87 Date: Fri, 5 Apr 2024 17:54:26 +0200 Subject: [PATCH] fix test --- test/components/generators/test_hugging_face_api.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/components/generators/test_hugging_face_api.py b/test/components/generators/test_hugging_face_api.py index 1bead5ffb3..697d2b6d78 100644 --- a/test/components/generators/test_hugging_face_api.py +++ b/test/components/generators/test_hugging_face_api.py @@ -39,7 +39,7 @@ def streaming_callback_handler(x): class TestHuggingFaceAPIGenerator: def test_init_invalid_api_type(self): with pytest.raises(ValueError): - HuggingFaceAPIGenerator(api_type="invalid_api_type") + HuggingFaceAPIGenerator(api_type="invalid_api_type", api_params={}) def test_init_serverless(self, mock_check_valid_model): model = "HuggingFaceH4/zephyr-7b-alpha"