rm pandas dependency (langchain-ai#2102)

kevinNejad · Mar 28, 2023 · f281033 · f281033
1 parent 410bf37
commit f281033
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 56 deletions.
diff --git a/langchain/document_loaders/dataframe.py b/langchain/document_loaders/dataframe.py
@@ -1,7 +1,5 @@
 """Load from Dataframe object"""
-from typing import List
-
-import pandas as pd
+from typing import Any, List
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader
@@ -10,8 +8,14 @@
 class DataFrameLoader(BaseLoader):
     """Load Pandas DataFrames."""
 
-    def __init__(self, data_frame: pd.DataFrame, page_content_column: str = "text"):
+    def __init__(self, data_frame: Any, page_content_column: str = "text"):
         """Initialize with dataframe object."""
+        import pandas as pd
+
+        if not isinstance(data_frame, pd.DataFrame):
+            raise ValueError(
+                f"Expected data_frame to be a pd.DataFrame, got {type(data_frame)}"
+            )
         self.data_frame = data_frame
         self.page_content_column = page_content_column
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,7 +17,6 @@ SQLAlchemy = "^1"
 requests = "^2"
 PyYAML = ">=5.4.1"
 numpy = "^1"
-pandas = ">=1.5.3"
 faiss-cpu = {version = "^1", optional = true}
 wikipedia = {version = "^1", optional = true}
 elasticsearch = {version = "^8", optional = true}

diff --git a/...t_tests/document_loader/test_dataframe.py → ..._tests/document_loaders/test_dataframe.py b/...t_tests/document_loader/test_dataframe.py → ..._tests/document_loaders/test_dataframe.py