Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Apr 5, 2024
1 parent 8253515 commit b8e9984
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/components/generators/test_hugging_face_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down

0 comments on commit b8e9984

Please sign in to comment.