feast/sdk/python/feast/field.py at master · codeboyyong/feast · GitHub

History

113 lines (93 loc) · 3.2 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# https://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from typing import Dict, Optional

from typeguard import typechecked

from feast.feature import Feature

from feast.protos.feast.core.Feature_pb2 import FeatureSpecV2 as FieldProto

from feast.types import FeastType, from_value_type

from feast.value_type import ValueType

@typechecked

class Field:

"""

A Field represents a set of values with the same structure.

Attributes:

dtype: The type of the field, such as string or float.

tags: User-defined metadata in dictionary form.

"""

dtype: FeastType

tags: Dict[str, str]

def __init__(

self,

dtype: FeastType,

tags: Optional[Dict[str, str]] = None,

"""

Creates a Field object.

Args:

dtype: The type of the field, such as string or float.

tags (optional): User-defined metadata in dictionary form.

"""

self.name = name

self.dtype = dtype

self.tags = tags or {}

def __eq__(self, other):

if type(self) != type(other):

return False

if (

self.name != other.name

or self.dtype != other.dtype

or self.tags != other.tags

return False

return True

def __hash__(self):

return hash((self.name, hash(self.dtype)))

def __lt__(self, other):

return self.name < other.name

def __repr__(self):

return f"{self.name}-{self.dtype}"

def __str__(self):

return f"Field(name={self.name}, dtype={self.dtype}, tags={self.tags})"

def to_proto(self) -> FieldProto:

"""Converts a Field object to its protobuf representation."""

value_type = self.dtype.to_value_type()

return FieldProto(name=self.name, value_type=value_type.value, tags=self.tags)

@classmethod

def from_proto(cls, field_proto: FieldProto):

"""

Creates a Field object from a protobuf representation.

Args:

field_proto: FieldProto protobuf object

"""

value_type = ValueType(field_proto.value_type)

return cls(

name=field_proto.name,

dtype=from_value_type(value_type=value_type),

tags=dict(field_proto.tags),

)

@classmethod

def from_feature(cls, feature: Feature):

"""

Creates a Field object from a Feature object.

Args:

feature: Feature object to convert.

"""

return cls(

name=feature.name, dtype=from_value_type(feature.dtype), tags=feature.labels

)

feast/sdk/python/feast/field.py at master · codeboyyong/feast · GitHub

Expand file tree

Breadcrumbs

field.py

Latest commit

History

Breadcrumbs

field.py

File metadata and controls