From df13880a9666bcbaac5ae35fabb7388f895f4615 Mon Sep 17 00:00:00 2001
From: Michal Szabados <michal@lightly.ai>
Date: Mon, 10 Mar 2025 16:12:43 +0100
Subject: [PATCH] Explicitly export input and output classes

---
 src/labelformat/formats/__init__.py | 30 +++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/src/labelformat/formats/__init__.py b/src/labelformat/formats/__init__.py
index f904994..3d6f840 100644
--- a/src/labelformat/formats/__init__.py
+++ b/src/labelformat/formats/__init__.py
@@ -47,3 +47,33 @@
     YOLOv11ObjectDetectionInput,
     YOLOv11ObjectDetectionOutput,
 )
+
+__all__ = [
+    "COCOInstanceSegmentationInput",
+    "COCOInstanceSegmentationOutput",
+    "COCOObjectDetectionInput",
+    "COCOObjectDetectionOutput",
+    "KittiObjectDetectionInput",
+    "KittiObjectDetectionOutput",
+    "LabelboxObjectDetectionInput",
+    "LightlyObjectDetectionInput",
+    "LightlyObjectDetectionOutput",
+    "PascalVOCObjectDetectionInput",
+    "PascalVOCObjectDetectionOutput",
+    "YOLOv5ObjectDetectionInput",
+    "YOLOv5ObjectDetectionOutput",
+    "YOLOv6ObjectDetectionInput",
+    "YOLOv6ObjectDetectionOutput",
+    "YOLOv7ObjectDetectionInput",
+    "YOLOv7ObjectDetectionOutput",
+    "YOLOv8InstanceSegmentationInput",
+    "YOLOv8InstanceSegmentationOutput",
+    "YOLOv8ObjectDetectionInput",
+    "YOLOv8ObjectDetectionOutput",
+    "YOLOv9ObjectDetectionInput",
+    "YOLOv9ObjectDetectionOutput",
+    "YOLOv10ObjectDetectionInput",
+    "YOLOv10ObjectDetectionOutput",
+    "YOLOv11ObjectDetectionInput",
+    "YOLOv11ObjectDetectionOutput",
+]