Fix tests

deepset-ai · Oct 10, 2024 · 8f6a464 · 8f6a464
1 parent 6cca5a7
commit 8f6a464
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/components/generators/test_openai_dalle.py b/test/components/generators/test_openai_dalle.py
@@ -10,7 +10,7 @@ def test_to_dict(self) -> None:
         generator = DALLEImageGenerator()
         data = generator.to_dict()
         assert data == {
-            "type": "dc_custom_component.components.generators.image_generator.DALLEImageGenerator",
+            "type": "haystack.components.generators.image_generator.DALLEImageGenerator",
             "init_parameters": {
                 "model": "dall-e-3",
                 "quality": "standard",
@@ -24,7 +24,7 @@ def test_to_dict(self) -> None:
 
     def test_from_dict(self) -> None:
         data = {
-            "type": "dc_custom_component.components.generators.image_generator.DALLEImageGenerator",
+            "type": "haystack.components.generators.image_generator.DALLEImageGenerator",
             "init_parameters": {
                 "model": "dall-e-3",
                 "quality": "standard",