fix: lint

wandb · Jan 22, 2025 · 39ccf64 · 39ccf64
1 parent 1866ba8
commit 39ccf64
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/weave/trace/autopatch.py b/weave/trace/autopatch.py
@@ -65,18 +65,16 @@ def autopatch(settings: Optional[AutopatchSettings] = None) -> None:
         get_google_genai_patcher,
     )
     from weave.integrations.groq.groq_sdk import get_groq_patcher
+    from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher
     from weave.integrations.instructor.instructor_sdk import get_instructor_patcher
-    from weave.integrations.langchain.langchain import langchain_patcher
     from weave.integrations.langchain_nvidia_ai_endpoints.langchain_nv_ai_endpoints import (
         get_nvidia_ai_patcher,
     )
     from weave.integrations.litellm.litellm import get_litellm_patcher
-    from weave.integrations.llamaindex.llamaindex import llamaindex_patcher
     from weave.integrations.mistral import get_mistral_patcher
     from weave.integrations.notdiamond.tracing import get_notdiamond_patcher
     from weave.integrations.openai.openai_sdk import get_openai_patcher
     from weave.integrations.vertexai.vertexai_sdk import get_vertexai_patcher
-    from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher
 
     get_openai_patcher(settings.openai).attempt_patch()
     get_mistral_patcher(settings.mistral).attempt_patch()
@@ -103,6 +101,7 @@ def reset_autopatch() -> None:
         get_google_genai_patcher,
     )
     from weave.integrations.groq.groq_sdk import get_groq_patcher
+    from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher
     from weave.integrations.instructor.instructor_sdk import get_instructor_patcher
     from weave.integrations.langchain.langchain import langchain_patcher
     from weave.integrations.langchain_nvidia_ai_endpoints.langchain_nv_ai_endpoints import (
@@ -114,7 +113,6 @@ def reset_autopatch() -> None:
     from weave.integrations.notdiamond.tracing import get_notdiamond_patcher
     from weave.integrations.openai.openai_sdk import get_openai_patcher
     from weave.integrations.vertexai.vertexai_sdk import get_vertexai_patcher
-    from weave.integrations.huggingface.huggingface_sdk import huggingface_patcher
 
     get_openai_patcher().undo_patch()
     get_mistral_patcher().undo_patch()