chore: refactor package files for types and chunkers

DS4SD · Sep 27, 2024 · 6c22aaf · 6c22aaf
1 parent 17af1fc
commit 6c22aaf
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 19 deletions.
diff --git a/docling_core/transforms/chunker/__init__.py b/docling_core/transforms/chunker/__init__.py
@@ -5,11 +5,7 @@
 
 """Define the chunker types."""
 
-from docling_core.transforms.chunker.base import (  # noqa
-    BaseChunker,
-    Chunk,
-    ChunkWithMetadata,
-)
-from docling_core.transforms.chunker.hierarchical_chunker import (  # noqa
-    HierarchicalChunker,
-)
+from docling_core.transforms.chunker.base import BaseChunker, Chunk, ChunkWithMetadata
+from docling_core.transforms.chunker.hierarchical_chunker import HierarchicalChunker
+
+__all__ = ["BaseChunker", "Chunk", "ChunkWithMetadata", "HierarchicalChunker"]
diff --git a/docling_core/types/__init__.py b/docling_core/types/__init__.py
@@ -5,21 +5,38 @@
 
 """Define the main types."""
 
-from docling_core.types.doc.base import BoundingBox  # noqa
-from docling_core.types.doc.base import Table  # noqa
-from docling_core.types.doc.base import TableCell  # noqa
-from docling_core.types.doc.base import (  # noqa
+from docling_core.types.doc.base import (
     BaseCell,
     BaseText,
+    BoundingBox,
     PageDimensions,
     PageReference,
     Prov,
     Ref,
+    Table,
+    TableCell,
 )
-from docling_core.types.doc.document import (  # noqa
+from docling_core.types.doc.document import (
     CCSDocumentDescription as DocumentDescription,
 )
-from docling_core.types.doc.document import CCSFileInfoObject as FileInfoObject  # noqa
-from docling_core.types.doc.document import ExportedCCSDocument as Document  # noqa
-from docling_core.types.gen.generic import Generic  # noqa
-from docling_core.types.rec.record import Record  # noqa
+from docling_core.types.doc.document import CCSFileInfoObject as FileInfoObject
+from docling_core.types.doc.document import ExportedCCSDocument as Document
+from docling_core.types.gen.generic import Generic
+from docling_core.types.rec.record import Record
+
+__all__ = [
+    "BaseCell",
+    "BaseText",
+    "BoundingBox",
+    "Document",
+    "DocumentDescription",
+    "FileInfoObject",
+    "Generic",
+    "PageDimensions",
+    "PageReference",
+    "Prov",
+    "Record",
+    "Ref",
+    "Table",
+    "TableCell",
+]
diff --git a/poetry.lock b/poetry.lock