fix tests

deepset-ai · Jan 17, 2024 · 248aaba · 248aaba
1 parent 9dd4964
commit 248aaba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/gradient/tests/test_gradient_text_embedder.py b/integrations/gradient/tests/test_gradient_text_embedder.py
@@ -59,7 +59,7 @@ def test_to_dict(self):
         data = component.to_dict()
         assert data == {
             "type": "gradient_haystack.embedders.gradient_text_embedder.GradientTextEmbedder",
-            "init_parameters": {"workspace_id": workspace_id, "model_name": "bge-large"},
+            "init_parameters": {"workspace_id": workspace_id, "model": "bge-large"},
         }
 
     @pytest.mark.unit