def METHOD_NAME(df, search_multi=False):
    """Find col that contains embed in pandas df """
    if not search_multi:
        for c in df.columns:
            if 'embed_entitiy' in c: return c
    else:
        e_cols = []
        for c in df.columns:
            if 'embed' in c: e_cols.append(c)
    return e_cols