Skip to content
Prev Previous commit
Next Next commit
Fix
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
kevjumba committed Apr 13, 2022
commit 512e00fe43a4a7c4b5a8e29e8a1b44faac908533
4 changes: 1 addition & 3 deletions sdk/python/feast/field.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,8 @@

from feast.feature import Feature
from feast.protos.feast.core.Feature_pb2 import FeatureSpecV2 as FieldProto
from feast.value_type import ValueType
from feast.types import FeastType, from_value_type
from feast.value_type import ValueType


class Field:
Expand Down Expand Up @@ -62,8 +62,6 @@ def __str__(self):

def to_proto(self) -> FieldProto:
"""Converts a Field object to its protobuf representation."""
if isinstance(self.dtype, ValueType):
return self.dtype.Enum
value_type = self.dtype.to_value_type()
return FieldProto(name=self.name, value_type=value_type.value)

Expand Down