b9dc15ecef0393a55a953acff3c1d0278b19d9f5,src/fonduer/utils/utils_udf.py,,get_sparse_matrix,#Any#Any#Any#Any#,31

Before Change

                cand_keys = values[0].keys
                cand_values = values[0].values
                indices.extend([keys.index(key) for key in cand_keys if key in keys])
                data.extend(
                    [cand_values[i[0]] for i in enumerate(cand_keys) if i[1] in keys]
                )

            indptr.append(len(indices))

        result.append(

After Change

        for cand in cand_list:
            values = _get_cand_values(cand, key_table)
            if values:
                for cand_key, cand_value in zip(values[0].keys, values[0].values):
                    if cand_key in keys_map:
                        indices.append(keys_map[cand_key])
                        data.append(cand_value)

            indptr.append(len(indices))

        result.append(
            csr_matrix((data, indices, indptr), shape=(len(cand_list), len(keys_map)))

Instances