Address review comments

Signed-off-by: Christoph Auer <[email protected]>
DS4SD · Feb 5, 2025 · 46ffd20 · 46ffd20
1 parent 2083941
commit 46ffd20
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/docling_core/types/doc/document.py b/docling_core/types/doc/document.py
@@ -63,6 +63,8 @@
     DocItemLabel.LIST_ITEM,
     DocItemLabel.CODE,
     DocItemLabel.REFERENCE,
+    DocItemLabel.PAGE_HEADER,
+    DocItemLabel.PAGE_FOOTER,
 }
 
 
@@ -513,6 +515,9 @@ class ContentLayer(str, Enum):
     FURNITURE = "furniture"
 
 
+DEFAULT_CONTENT_LAYERS = {ContentLayer.BODY}
+
+
 class NodeItem(BaseModel):
     """NodeItem."""
 
@@ -1851,7 +1856,7 @@ def iterate_items(
         with_groups: bool = False,
         traverse_pictures: bool = False,
         page_no: Optional[int] = None,
-        included_content_layers: List[ContentLayer] = [ContentLayer.BODY],
+        included_content_layers: set[ContentLayer] = DEFAULT_CONTENT_LAYERS,
         _level: int = 0,  # fixed parameter, carries through the node nesting level
     ) -> typing.Iterable[Tuple[NodeItem, int]]:  # tuple of node and level
         """iterate_elements.