(orig_pipeline)
return result
Wrap segments of the pipeline to mark them for pretty_print() and visualize().

    If the pipeline does not look like it came from AutoAI, just return it
    unchanged. Otherwise, find the NumpyPermuteArray operator. Everything
    before that operator is preprocessing. Everything after
    NumpyPermuteArray but before the final estimator is feature
    engineering.
Wrap segments of the pipeline to mark them for pretty_print() and visualize().
Wrap segments of the pipeline to mark them for pretty_print () and visualize () .
def wrap_pipeline_segments(orig_pipeline):
    """Wrap segments of the pipeline to mark them for pretty_print() and visualize().

    If the pipeline does not look like it came from AutoAI, just return it
    unchanged. Otherwise, find the NumpyPermuteArray operator. Everything
    before that operator is preprocessing. Everything after
    NumpyPermuteArray but before the final estimator is feature
    engineering."""
    from lale.lib.autoai_libs.numpy_permute_array import NumpyPermuteArray

    if len(orig_pipeline.steps()) <= 2:
        return orig_pipeline
    estimator = orig_pipeline.get_last()
    prep = orig_pipeline.remove_last()
    cognito = None
    PREP_END = NumpyPermuteArray.class_name()
    while True:
        last = prep.get_last()
        if last is None or not last.class_name().startswith("lale.lib.autoai_libs."):
            return orig_pipeline
        if last.class_name() == PREP_END:
            break
        prep = prep.remove_last()
        if cognito is None:
            cognito = last
        else:
            cognito = last >> cognito
    prep_wrapped = lale.lib.sklearn.Pipeline(steps=[("preprocessing_pipeline", prep)])
    if cognito is None:
        result = prep_wrapped >> estimator
    else:
        cognito_wrapped = lale.lib.sklearn.Pipeline(
            steps=[("feature_engineering_pipeline", cognito)]
        )
        result = prep_wrapped >> cognito_wrapped >> estimator
    return result
def wrap_pipeline_segments ( orig_pipeline ) : from lale . lib . autoai_libs . numpy_permute_array import NumpyPermuteArray if len ( orig_pipeline . steps ( ) ) <= 2 : return orig_pipeline estimator = orig_pipeline . get_last ( ) prep = orig_pipeline . remove_last ( ) cognito = None PREP_END = NumpyPermuteArray . class_name ( ) while True : last = prep . get_last ( ) if last is None or not last . class_name ( ) . startswith ( "lale.lib.autoai_libs." ) : return orig_pipeline if last . class_name ( ) == PREP_END : break prep = prep . remove_last ( ) if cognito is None : cognito = last else : cognito = last >> cognito prep_wrapped = lale . lib . sklearn . Pipeline ( steps = [ ( "preprocessing_pipeline" , prep ) ] ) if cognito is None : result = prep_wrapped >> estimator else : cognito_wrapped = lale . lib . sklearn . Pipeline ( steps = [ ( "feature_engineering_pipeline" , cognito ) ] ) result = prep_wrapped >> cognito_wrapped >> estimator return result