Ensure cell block has text element

aws-samples · May 6, 2024 · 44a002d · 44a002d
2 parents 54aeaca + 8d5d817
commit 44a002d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prettyprinter/textractprettyprinter/t_pretty_print_layout.py b/prettyprinter/textractprettyprinter/t_pretty_print_layout.py
@@ -108,7 +108,7 @@ def _dfs(self, root, id2block):
                             for cell_id in cell_rel['Ids']:
                                 cell_block = id2block[cell_id]
                                 if "Relationships" in cell_block:
-                                    cell_text = " ".join([id2block[line_id]['Text'] for line_id in cell_block["Relationships"][0]['Ids']])
+                                    cell_text = " ".join([id2block[line_id]['Text'] for line_id in cell_block["Relationships"][0]['Ids'] if 'Text' in id2block[line_id]])
                                     row_idx = cell_block['RowIndex']
                                     col_idx = cell_block['ColumnIndex']
                                     max_row = max(max_row, row_idx)