Skip to content
Prev Previous commit
Next Next commit
Fix tests
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
  • Loading branch information
kevjumba committed Apr 13, 2022
commit 156b171a826d7cc85c60d4c089c492d508c1aa1f
2 changes: 1 addition & 1 deletion protos/feast/types/Field.proto
Original file line number Diff line number Diff line change
Expand Up @@ -26,5 +26,5 @@ option go_package = "github.com/feast-dev/feast/go/protos/feast/types";

message Field {
string name = 1;
feast.types.Value value = 2;
feast.types.ValueType.Enum value = 2;
}
4 changes: 2 additions & 2 deletions sdk/python/feast/field.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@

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


Expand Down Expand Up @@ -76,7 +76,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))
return cls(name=field_proto.name, dtype=from_value_type(value_type=value_type))

@classmethod
def from_feature(cls, feature: Feature):
Expand Down