Fix table indexing returning the entire table

aws-samples · Aug 20, 2024 · e608576 · e608576
1 parent 5764d4b
commit e608576
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/textractor/entities/table.py b/textractor/entities/table.py
@@ -490,7 +490,8 @@ def __getitem__(self, key):
         new_table_cells = _get_new_table_cells(rows, filtered_rows)
 
         new_table = deepcopy(self)
-        new_table.add_cells(new_table_cells)
+        new_table.table_cells = new_table_cells
+        new_table._children = new_table_cells
 
         return new_table