Function _build_entity_keys

sdk/python/feast/utils.py:384–407

Build entity key protos for each row. Args: num_rows: Number of rows to generate entity keys for. join_keys: Dict mapping join key names to their value types. proto_values: Dict mapping column names to lists of ValueProto values. Returns: List of EntityKeyPr

(
    num_rows: int,
    join_keys: Dict[str, ValueType],
    proto_values: Dict[str, List[ValueProto]],
)

def _build_entity_keys(
    num_rows: int,
    join_keys: Dict[str, ValueType],
    proto_values: Dict[str, List[ValueProto]],
) -> List[EntityKeyProto]:
    """Build entity key protos for each row.

    Args:
        num_rows: Number of rows to generate entity keys for.
        join_keys: Dict mapping join key names to their value types.
        proto_values: Dict mapping column names to lists of ValueProto values.

    Returns:
        List of EntityKeyProto, one per row.
    """
    return [
        EntityKeyProto(
            join_keys=list(join_keys.keys()),
            entity_values=[
                proto_values[k][idx] for k in join_keys if k in proto_values
            ],
        )
        for idx in range(num_rows)
    ]


def _convert_arrow_to_proto(