Skip to content

fix: Update field api to add tag parameter corresponding to labels in Feature. #2610

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
feast-ci-bot merged 12 commits into from
Apr 26, 2022
Merged

fix: Update field api to add tag parameter corresponding to labels in Feature. #2610

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
8e10875
Fix tags
kevjumba Apr 25, 2022
3b8c38d
Fix lint
kevjumba Apr 25, 2022
b9b6bbb
Fix lint
kevjumba Apr 25, 2022
ce5753e
Fix lint
kevjumba Apr 25, 2022
4a83454
Convert field to use optional kw args
kevjumba Apr 25, 2022
0a0e420
fix
kevjumba Apr 25, 2022
5eeb0d4
Fix
kevjumba Apr 25, 2022
ccf6e9b
Fix
kevjumba Apr 25, 2022
fab96ec
Fix
kevjumba Apr 26, 2022
3654ebb
Fix
kevjumba Apr 26, 2022
7d3975f
Fix lint
kevjumba Apr 26, 2022
fbf27be
Fix java tests
kevjumba Apr 26, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Fix tags 
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
@kevjumba
kevjumba committed Apr 25, 2022
commit 8e10875eab0bbfa47d09ca807c13cbfa42f7e10a
3 changes: 3 additions & 0 deletions protos/feast/types/Field.proto
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,4 +27,7 @@ option go_package = "github.com/feast-dev/feast/go/protos/feast/types";
message Field {
string name = 1;
feast.types.ValueType.Enum value = 2;

// User defined metadata
map<string,string> tags = 3;
Comment thread
felixwang9817 marked this conversation as resolved.
Outdated
}
22 changes: 15 additions & 7 deletions sdk/python/feast/field.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,10 @@
# See the License for the specific language governing permissions and
# limitations under the License.

from typing import Dict, Optional

from attr import field

from feast.feature import Feature
from feast.protos.feast.core.Feature_pb2 import FeatureSpecV2 as FieldProto
from feast.types import FeastType, from_value_type
Expand All @@ -25,31 +29,35 @@ class Field:
Attributes:
name: The name of the field.
dtype: The type of the field, such as string or float.
tags: User-defined metadata in dictionary form.
"""

name: str
dtype: FeastType
tags: Dict[str, str]

def __init__(
self, *, name: str, dtype: FeastType,
self, *, name: str, dtype: FeastType, tags: Optional[Dict[str, str]] = None,
):
"""
Creates a Field object.

Args:
name: The name of the field.
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 self.name != other.name or self.dtype != other.dtype:
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)))
return hash((self.name, hash(self.dtype), hash(self.tags)))
Comment thread
felixwang9817 marked this conversation as resolved.
Outdated

def __lt__(self, other):
return self.name < other.name
Expand All @@ -58,12 +66,12 @@ def __repr__(self):
return f"{self.name}-{self.dtype}"

def __str__(self):
return f"Field(name={self.name}, dtype={self.dtype})"
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)
return FieldProto(name=self.name, value_type=value_type.value, tags=self.tags)

@classmethod
def from_proto(cls, field_proto: FieldProto):
Expand All @@ -74,7 +82,7 @@ def from_proto(cls, field_proto: FieldProto):
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))
return cls(name=field_proto.name, dtype=from_value_type(value_type=value_type), tags=field_proto.tags)

@classmethod
def from_feature(cls, feature: Feature):
Expand All @@ -84,4 +92,4 @@ def from_feature(cls, feature: Feature):
Args:
feature: Feature object to convert.
"""
return cls(name=feature.name, dtype=from_value_type(feature.dtype))
return cls(name=feature.name, dtype=from_value_type(feature.dtype), tags=feature.labels)