a3400845afb88d0d9b19f7f810abbd718117f24e,src/fonduer/utils/utils_udf.py,,get_sparse_matrix,#Any#Any#Any#Any#,102

Before Change

    if key:
        keys_map = {key: 0}
    else:
        keys_map = {
            key.name: idx
            for (idx, key) in enumerate(get_sparse_matrix_keys(session, key_table))
        }

    for cand_list in cand_lists:
        indptr = [0]
        indices = []

After Change

    cand_lists = cand_lists if isinstance(cand_lists, (list, tuple)) else [cand_lists]

    for cand_list in cand_lists:
        if len(cand_list) == 0:
            raise ValueError("cand_lists contain empty cand_list.")
        candidate_class = cand_list[0].__tablename__

        // Keys are used as a global index
        if key:
            keys_map = {key: 0}

Instances