Change type of PromptModel invocation_layer_class init param (#6497)

deepset-ai · Dec 15, 2023 · fdee07f · fdee07f
1 parent 9afa66a
commit fdee07f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/prompt/test_prompt_model.py b/test/prompt/test_prompt_model.py
@@ -40,7 +40,7 @@ def test_constructor_with_no_supported_model():
 
 
 @pytest.mark.unit
-def test_constructor_with_invocation_layer_class_string(mock_auto_tokenizer):
+def test_constructor_with_invocation_layer_class_string():
     model = PromptModel(
         invocation_layer_class="haystack.nodes.prompt.invocation_layer.CohereInvocationLayer", api_key="fake_api_key"
     )