ea7d11a94cd5722f530d7eff2139f499f7bdd1a6,tpot/export_utils.py,,generate_import_code,#,183

Before Change

    // Build dict of import requirments from list of operators
    import_relations = {op.__name__: op.import_hash for op in operators}
    print(import_relations)  //// to be removed on merge

    if "imblearn" in import_relations.values():
        pipeline_module = "imblearn"

After Change

    import_relations = {op.__name__: op.import_hash for op in operators}

    flatten_list = lambda list_: [item for sublist in list_ for item in sublist]
    modules_used = [
        module.split(".")[0]
        for module in flatten_list(
            [list(val.keys()) for val in import_relations.values()]
        )
    ]

    if "imblearn" in modules_used:
        pipeline_module = "imblearn"
    else:

Instances