align usage of content layer param

Signed-off-by: Panos Vagenas <[email protected]>

Author: vagenas

docling_core/transforms/visualizer/layout_visualizer.py

@@ -141,9 +141,7 @@ def _draw_doc_layout(
         prev_image = None
         prev_page_nr = None
         for idx, (elem, _) in enumerate(
-            doc.iterate_items(
-                included_content_layers={ContentLayer.BODY, ContentLayer.FURNITURE}
-            )
+            doc.iterate_items(included_content_layers={cl for cl in ContentLayer})
         ):
             if not isinstance(elem, DocItem):
                 continue

docling_core/transforms/visualizer/reading_order_visualizer.py

@@ -71,7 +71,7 @@ def _draw_doc_reading_order(
         my_images: dict[Optional[int], Image] = images or {}
         prev_page = None
         for elem, _ in doc.iterate_items(
-            included_content_layers={ContentLayer.BODY, ContentLayer.FURNITURE},
+            included_content_layers={cl for cl in ContentLayer},
         ):
             if not isinstance(elem, DocItem):
                 continue

docling_core/types/doc/document.py

@@ -2873,7 +2873,7 @@ def print_element_tree(self):
             self.iterate_items(
                 with_groups=True,
                 traverse_pictures=True,
-                included_content_layers={ContentLayer.BODY, ContentLayer.FURNITURE},
+                included_content_layers={cl for cl in ContentLayer},
             )
         ):
             if isinstance(item, GroupItem):
@@ -2897,7 +2897,7 @@ def export_to_element_tree(self) -> str:
             self.iterate_items(
                 with_groups=True,
                 traverse_pictures=True,
-                included_content_layers={ContentLayer.BODY, ContentLayer.FURNITURE},
+                included_content_layers={cl for cl in ContentLayer},
             )
         ):
             if isinstance(item, GroupItem):