From 6b14d9b3401bd632592104d9a94dd38d79817c17 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Tue, 16 Jul 2019 09:50:35 +0200
Subject: [PATCH] update init

---
 pts/feature/__init__.py  | 27 ++++++++++++++++++++++++++-
 pts/feature/transform.py |  2 +-
 2 files changed, 27 insertions(+), 2 deletions(-)

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]: