fixed unit test

SeldonIO · Oct 5, 2023 · b479d79 · b479d79
1 parent cdc5ec8
commit b479d79
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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()