fix: loosen pandas version (#40)

* fix: loosen pandas version Signed-off-by: Panos Vagenas <[email protected]> * add pandas stubs Signed-off-by: Panos Vagenas <[email protected]> * explicitly convert to str Signed-off-by: Panos Vagenas <[email protected]> --------- Signed-off-by: Panos Vagenas <[email protected]>
DS4SD · Oct 9, 2024 · aec1a41 · aec1a41
1 parent 345419f
commit aec1a41
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 66 deletions.
diff --git a/docling_core/transforms/chunker/hierarchical_chunker.py b/docling_core/transforms/chunker/hierarchical_chunker.py
@@ -83,7 +83,7 @@ def _triplet_serialize(cls, table) -> Optional[str]:
             nrows = table_df.shape[0]
             ncols = table_df.shape[1]
             texts = [
-                f"{rows[i]}, {cols[j]} = {table_df.iloc[i, j].strip()}"
+                f"{rows[i]}, {cols[j]} = {str(table_df.iloc[i, j]).strip()}"
                 for i in range(1, nrows)
                 for j in range(1, ncols)
             ]

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -52,7 +52,7 @@ pydantic = "^2.6.0"
 jsonref = "^1.1.0"
 json-schema-for-humans = "^1.0.0"
 tabulate = "^0.9.0"
-pandas = "^2.2.2"
+pandas = "^2.1.4"
 
 [tool.poetry.group.dev.dependencies]
 black = "^24.4.2"
@@ -68,6 +68,7 @@ pep8-naming = "^0.13.2"
 jsondiff = "^2.0.0"
 types-setuptools = "^70.3.0"
 python-semantic-release = "^7.32.2"
+pandas-stubs = "^2.1.4.231227"
 
 [tool.setuptools.packages.find]
 where = ["docling_core/resources/schemas"]
@@ -117,7 +118,6 @@ module = [
     "jsonref.*",
     "jsonschema.*",
     "json_schema_for_humans.*",
-    "pandas.*",
     "requests.*",
     "tabulate.*",
     "yaml.*",