fix leftover model_name_or_path param

deepset-ai · Jan 12, 2024 · cdd5e7f · cdd5e7f
1 parent 24c71bd
commit cdd5e7f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/e2e/pipelines/test_eval_rag_pipelines.py b/e2e/pipelines/test_eval_rag_pipelines.py
@@ -28,7 +28,7 @@ def test_bm25_rag_pipeline(tmp_path):
     rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder")
     rag_pipeline.add_component(
         instance=HuggingFaceLocalGenerator(
-            model_name_or_path="google/flan-t5-small",
+            model="google/flan-t5-small",
             task="text2text-generation",
             generation_kwargs={"max_new_tokens": 100, "temperature": 0.5, "do_sample": True},
         ),
@@ -102,7 +102,7 @@ def test_embedding_retrieval_rag_pipeline(tmp_path):
     rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name="prompt_builder")
     rag_pipeline.add_component(
         instance=HuggingFaceLocalGenerator(
-            model_name_or_path="google/flan-t5-small",
+            model="google/flan-t5-small",
             task="text2text-generation",
             generation_kwargs={"max_new_tokens": 100, "temperature": 0.5, "do_sample": True},
         ),