fixed typo

SeldonIO · Oct 4, 2023 · cdc5ec8 · cdc5ec8
1 parent d16b0e1
commit cdc5ec8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/runtimes/huggingface/tests/test_common.py b/runtimes/huggingface/tests/test_common.py
@@ -155,7 +155,7 @@ def test_pipeline_is_initialised_with_correct_model_kwargs(
 def test_pipeline_uses_model_kwargs(
     pretrained_model: str,
     model_kwargs: Optional[dict],
-    expected: bool,
+    expected: torch.dtype,
 ):
     hf_settings = HuggingFaceSettings(
         pretrained_model=pretrained_model,