Function _columns_to_proto_values

sdk/python/feast/utils.py:358–381

Convert table columns to proto values dict. Args: table: PyArrow RecordBatch containing the data. columns: List of (column_name, value_type) tuples to convert. allow_missing: If True, skip columns not found in table. If False, raise ValueError. Returns: Dict

(
    table: pyarrow.RecordBatch,
    columns: List[Tuple[str, ValueType]],
    allow_missing: bool = False,
)

def _columns_to_proto_values(
    table: pyarrow.RecordBatch,
    columns: List[Tuple[str, ValueType]],
    allow_missing: bool = False,
) -> Dict[str, List[ValueProto]]:
    """Convert table columns to proto values dict.

    Args:
        table: PyArrow RecordBatch containing the data.
        columns: List of (column_name, value_type) tuples to convert.
        allow_missing: If True, skip columns not found in table. If False, raise ValueError.

    Returns:
        Dict mapping column names to lists of ValueProto.
    """
    result: Dict[str, List[ValueProto]] = {}
    for column, value_type in columns:
        if column in table.column_names:
            result[column] = python_values_to_proto_values(
                table.column(column).to_numpy(zero_copy_only=False), value_type
            )
        elif not allow_missing:
            raise ValueError(f"Column {column} not found in table")
    return result


def _build_entity_keys(