updated feature union class

Author: yuanchi2807

codeflare/pipelines/tests/test_and.py

@@ -9,10 +9,13 @@
 from codeflare.pipelines.Datamodel import XYRef
 from codeflare.pipelines.Runtime import ExecutionType
 
-class FeatureUnion(dm.AndTransform):
+class FeatureUnion(dm.AndEstimator):
     def __init__(self):
         pass
 
+    def get_estimator_type(self):
+        return 'transform'
+
     def transform(self, xy_list):
         X_list = []
         y_vec = None

codeflare/pipelines/tests/test_multibranch.py

@@ -14,7 +14,7 @@
 from codeflare.pipelines.Datamodel import XYRef
 from codeflare.pipelines.Runtime import ExecutionType
 
-class FeatureUnion(dm.AndTransform):
+class FeatureUnion(dm.AndEstimator):
     def __init__(self):
         pass