Fix patch import

argilla-io · Oct 25, 2024 · 36361b7 · 36361b7
1 parent 1c1b74a
commit 36361b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/distilabel/models/llms/azure.py b/src/distilabel/models/llms/azure.py
@@ -122,7 +122,7 @@ def load(self) -> None:
         # This is a workaround to avoid the `OpenAILLM` calling the _prepare_structured_output
         # in the load method before we have the proper client.
         with patch(
-            "distilabel.llms.openai.OpenAILLM._prepare_structured_output", lambda x: x
+            "distilabel.models.openai.OpenAILLM._prepare_structured_output", lambda x: x
         ):
             super().load()