Pre-commit

Author: dtronmans

luxonis_ml/data/datasets/luxonis_dataset.py

@@ -17,8 +17,7 @@
 from semver.version import Version
 from typing_extensions import Self, override
 
-from luxonis_ml.data.exporters import BaseExporter, NativeExporter
-from luxonis_ml.data.exporters.prepared_ldf import PreparedLDF
+from luxonis_ml.data.exporters import BaseExporter, NativeExporter, PreparedLDF
 from luxonis_ml.data.utils import (
     BucketStorage,
     BucketType,

luxonis_ml/data/exporters/__init__.py

@@ -1,4 +1,5 @@
 from .base_exporter import BaseExporter
 from .native_exporter import NativeExporter
+from .prepared_ldf import PreparedLDF
 
-__all__ = ["BaseExporter", "NativeExporter"]
+__all__ = ["BaseExporter", "NativeExporter", "PreparedLDF"]

luxonis_ml/data/exporters/prepared_ldf.py

@@ -1,9 +1,10 @@
 from pathlib import Path
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 import polars as pl
 
-from luxonis_ml.data import LuxonisDataset
+if TYPE_CHECKING:
+    from luxonis_ml.data.datasets.luxonis_dataset import LuxonisDataset
 
 
 class PreparedLDF:
@@ -20,7 +21,7 @@ def __init__(
         self.grouped_image_sources = grouped_image_sources
 
     @classmethod
-    def from_dataset(cls, ldf: LuxonisDataset) -> "PreparedLDF":
+    def from_dataset(cls, ldf: "LuxonisDataset") -> "PreparedLDF":
         """Prepare a dataset for export into the LDF representation."""
         splits = ldf.get_splits()
         if splits is None: