diff --git a/pts/feature/__init__.py b/pts/feature/__init__.py
index 700f29d..ca0e662 100644
--- a/pts/feature/__init__.py
+++ b/pts/feature/__init__.py
@@ -8,4 +8,29 @@ from pts.feature.time_feature import (
     WeekOfYear,
 )
 
-from pts.feature.transform import Transformation, Chain
+from pts.feature.transform import (
+    Transformation,
+    Chain,
+    IdentityTransformation,
+    MapTransformation,
+    SimpleTransformation,
+    AdhocTransform,
+    FlatMapTransformation,
+    FilterTransformation,
+    RemoveFields,
+    SetField,
+    AsNumpyArray,
+    ExpandDimArray,
+    VstackFeatures,
+    ConcatFeatures,
+    SwapAxes,
+    ListFeatures,
+    AddObservedValuesIndicator,
+    RenameFields,
+    AddConstFeature,
+    AddTimeFeatures,
+    AddAgeFeature,
+    InstanceSplitter,
+    CanonicalInstanceSplitter,
+    SelectFields,
+)
diff --git a/pts/feature/transform.py b/pts/feature/transform.py
index f8c20a8..4b6fcb7 100644
--- a/pts/feature/transform.py
+++ b/pts/feature/transform.py
@@ -64,7 +64,7 @@ class Chain(Transformation):
         return reduce(lambda x, y: y.estimate(x), self.trans, data_it)
 
 
-class Identity(Transformation):
+class IdentityTransformation(Transformation):
     def __call__(
             self, data_it: Iterator[DataEntry], is_train: bool
     ) -> Iterator[DataEntry]: