from tfx.components import Transform

def preprocessing_fn(inputs):
    # Normalize the numeric features
    outputs = {
        'sepal_length': Transform.scale_to_z_score(inputs['sepal.length']),
        'sepal_width': Transform.scale_to_z_score(inputs['sepal.width']),
        'petal_length': Transform.scale_to_z_score(inputs['petal.ength']),
        'petal_width': Transform.scale_to_z_score(inputs['petal.width']),
        'species': inputs['variety']
    }
    return outputs

transform = Transform(
    examples=example_gen.outputs['examples'],
    schema=schema_gen.outputs['schema'],
    module_file='/content/preprocessing.py'
)

        __
        __