Lint files

Sheepsta300 · Oct 8, 2024 · 372a5d2 · 372a5d2
1 parent 15b137d
commit 372a5d2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/docs/docs/integrations/tools/azure_ai_face.ipynb b/docs/docs/integrations/tools/azure_ai_face.ipynb
@@ -95,7 +95,7 @@
    "source": [
     "from langchain.agents import AgentExecutor, create_structured_chat_agent\n",
     "from langchain_community.tools.azure_ai_services.face_analysis import (\n",
-    "    AzureAIFaceAnalysisTool\n",
+    "    AzureAIFaceAnalysisTool,\n",
     ")\n",
     "from langchain_openai import AzureChatOpenAI"
    ]

diff --git a/libs/community/tests/unit_tests/tools/azure_ai_services/test_face_analysis.py b/libs/community/tests/unit_tests/tools/azure_ai_services/test_face_analysis.py
@@ -36,8 +36,8 @@ def test_faces_detected(mocker: Any) -> None:
     key = "key"
     endpoint = "endpoint"
 
+    mocker.patch("azure.ai.vision.face.FaceClient", autospec=True)
     mocker.patch("azure.core.credentials.AzureKeyCredential", autospec=True)
-    mocker.patch("azure.ai.contentsafety.ContentSafetyClient", autospec=True)
     tool = AzureAIFaceAnalysisTool(
         azure_ai_face_key=key, azure_ai_face_endpoint=endpoint
     )
@@ -61,8 +61,8 @@ def test_no_faces_detected(mocker: Any) -> None:
     key = "key"
     endpoint = "endpoint"
 
+    mocker.patch("azure.ai.vision.face.FaceClient", autospec=True)
     mocker.patch("azure.core.credentials.AzureKeyCredential", autospec=True)
-    mocker.patch("azure.ai.contentsafety.ContentSafetyClient", autospec=True)
     tool = AzureAIFaceAnalysisTool(
         azure_ai_face_key=key, azure_ai_face_endpoint=endpoint
     )