diff --git a/atomsci/ddm/pipeline/featurization.py b/atomsci/ddm/pipeline/featurization.py index dc05d886..fe0eb3db 100644 --- a/atomsci/ddm/pipeline/featurization.py +++ b/atomsci/ddm/pipeline/featurization.py @@ -1606,7 +1606,7 @@ def create_feature_transformer(self, dataset, params): """ if params.feature_transform_type == 'normalization': transformers_x = [trans.NormalizationTransformerMissingData(transform_X=True, dataset=dataset)] - elif params.feature_transformer_type == 'RobustScaler': + elif params.feature_transform_type == 'RobustScaler': transformers_x = [ trans.SklearnTransformerWrapper(transform_X=True, dataset=dataset, sklearn_transformer=RobustScaler( @@ -1616,7 +1616,7 @@ def create_feature_transformer(self, dataset, params): unit_variance=params.robustscaler_unit_variance )) ] - elif params.feature_transformer_type == 'PowerTransformer': + elif params.feature_transform_type == 'PowerTransformer': transformers_x = [ trans.SklearnTransformerWrapper(transform_X=True, dataset=dataset, sklearn_transformer=PowerTransformer(