fix: correct patch target for _prepare_structured_output in AzureOpen…

…AILLM.load
argilla-io · Mar 2, 2025 · 1e4f0c9 · 1e4f0c9
1 parent 1b6c101
commit 1e4f0c9
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.models.openai.OpenAILLM._prepare_structured_output", lambda x: x
+            "distilabel.models.llms.openai.OpenAILLM._prepare_structured_output", lambda x: x
         ):
             super().load()