yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-23 05:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix table extraction in ParsrConverter (#2262)

Browse Source
This commit is contained in:
bogdankostic 2022-03-02 09:21:54 +01:00 committed by GitHub
parent c5542bd3fb
commit a5501c72cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
haystack/nodes/file_converter/parsr.py
View File

@ -197,9 +197,12 @@ class ParsrConverter(BaseConverter):
elem_idx: int, elem_idx: int,
meta: Optional[Dict[str, str]] = None, meta: Optional[Dict[str, str]] = None,
) -> Dict[str, Any]: ) -> Dict[str, Any]:
row_idx_start = 0 row_idx_start = 0
caption = "" caption = ""
table_list = [[""] * len(element["content"][0]["content"]) for _ in range(len(element["content"]))] number_of_columns = max([len(row["content"]) for row in element["content"]])
number_of_rows = len(element["content"])
table_list = [[""] * number_of_columns for _ in range(number_of_rows)]
for row_idx, row in enumerate(element["content"]): for row_idx, row in enumerate(element["content"]):
for col_idx, cell in enumerate(row["content"]): for col_idx, cell in enumerate(row["content"]):