diff --git a/libs/transforms/src/transforms/api/__init__.py b/libs/transforms/src/transforms/api/__init__.py
index dabf38b..8c7958c 100644
--- a/libs/transforms/src/transforms/api/__init__.py
+++ b/libs/transforms/src/transforms/api/__init__.py
@@ -15,7 +15,7 @@
     transform_pandas,
     transform_polars,
 )
-from transforms.api._transform import Transform, TransformContext
+from transforms.api._transform import Transform, TransformContext, TransformInput, TransformOutput
 
 __all__ = (
     "Input",
@@ -31,4 +31,6 @@
     "transform_polars",
     "Transform",
     "TransformContext",
+    "TransformInput",
+    "TransformOutput",
 )