Corrected AzureOpenAILLM path to mock the internal class (#1115)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
argilla-io · Feb 18, 2025 · 2d4ec9a · 2d4ec9a
1 parent 0009d1c
commit 2d4ec9a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/distilabel/models/llms/azure.py b/src/distilabel/models/llms/azure.py
@@ -122,7 +122,8 @@ 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()