Class Array

sdk/python/feast/types.py:177–218

An Array represents a list of types. Attributes: base_type: The base type of the array.

class Array(ComplexFeastType):
    """
    An Array represents a list of types.

    Attributes:
        base_type: The base type of the array.
    """

    base_type: Union[PrimitiveFeastType, ComplexFeastType]

    def __init__(self, base_type: Union[PrimitiveFeastType, "ComplexFeastType"]):
        # Allow Struct, Array, and Set as base types for nested collections
        if (
            not isinstance(base_type, (Struct, Array, Set))
            and base_type not in SUPPORTED_BASE_TYPES
        ):
            raise ValueError(
                f"Type {type(base_type)} is currently not supported as a base type for Array."
            )

        self.base_type = base_type

    def to_value_type(self) -> ValueType:
        if isinstance(self.base_type, Struct):
            return ValueType.STRUCT_LIST
        if isinstance(self.base_type, (Array, Set)):
            return ValueType.VALUE_LIST
        assert isinstance(self.base_type, PrimitiveFeastType)
        value_type_name = PRIMITIVE_FEAST_TYPES_TO_VALUE_TYPES[self.base_type.name]
        value_type_list_name = value_type_name + "_LIST"
        return ValueType[value_type_list_name]

    def __eq__(self, other):
        if isinstance(other, Array):
            return self.base_type == other.base_type
        return False

    def __hash__(self):
        return hash(("Array", hash(self.base_type)))

    def __str__(self):
        return f"Array({self.base_type})"


class Set(ComplexFeastType):