From b479d798167f6bf03eefe701245d7e6468f531c8 Mon Sep 17 00:00:00 2001 From: Nanbo Liu Date: Thu, 5 Oct 2023 15:19:14 +0000 Subject: [PATCH] fixed unit test --- runtimes/huggingface/tests/test_common.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/runtimes/huggingface/tests/test_common.py b/runtimes/huggingface/tests/test_common.py index aa6ad95bc..d7b60c51d 100644 --- a/runtimes/huggingface/tests/test_common.py +++ b/runtimes/huggingface/tests/test_common.py @@ -123,12 +123,10 @@ def test_pipeline_is_initialised_with_correct_model_kwargs( mock_pipeline_factory.return_value = MagicMock() hf_settings = HuggingFaceSettings(model_kwargs=model_kwargs) - + model_params = ModelParameters(uri="dummy_uri") model_settings = ModelSettings( - name="foo", - implementation=HuggingFaceRuntime, + name="foo", implementation=HuggingFaceRuntime, parameters=model_params ) - _ = load_pipeline_from_settings(hf_settings, model_settings) mock_pipeline_factory.return_value.assert_called_once()