From 5ccebce777e3a0e9b281e87892d83efcb9fb8a96 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Sun, 18 Jun 2023 18:33:04 -0700 Subject: [PATCH] rm pandas from arize (#6392) --- langchain/callbacks/arize_callback.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/langchain/callbacks/arize_callback.py b/langchain/callbacks/arize_callback.py index c1b4106f..a93dddfe 100644 --- 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 @@ class ArizeCallbackHandler(BaseCallbackHandler): 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