rm pandas from arize (#6392)

langchain-ai · Jun 19, 2023 · 5ccebce · 5ccebce
1 parent 3b7c4c5
commit 5ccebce
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/langchain/callbacks/arize_callback.py b/langchain/callbacks/arize_callback.py
@@ -2,8 +2,6 @@
 import uuid
 from typing import Any, Dict, List, Optional, Union
 
-import pandas as pd
-
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction, AgentFinish, LLMResult
 
@@ -77,6 +75,7 @@ def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
     def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
         """Log data to Arize when an LLM ends."""
 
+        import pandas as pd
         from arize.utils.types import Embedding, Environments, ModelTypes
 
         # Record token usage of the LLM