def get_tool()

in Project-AutoML/train.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

def get_tool(tool_name):
    tool_name = tool_name.lower()
    assert tool_name in {"autosklearn", "flaml"}
    if tool_name.lower() == "autosklearn":
        from automl.mod.mod_autosklearn import AutoSklearn as Tool

    elif tool_name.lower() == "flaml":
        from automl.mod.mod_flaml import FLAML as Tool

    else:
        raise Exception(f"Does not support {tool_name}")
    return Tool