From fb988ed74ff22a3937eaa9a1447018c7299988c8 Mon Sep 17 00:00:00 2001 From: hkuepers Date: Tue, 25 Feb 2025 17:08:15 +0100 Subject: [PATCH] Set value_type of entity directly in from_proto Signed-off-by: hkuepers --- sdk/python/feast/entity.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sdk/python/feast/entity.py b/sdk/python/feast/entity.py index 9c529115c8e..7f4eadc6352 100644 --- a/sdk/python/feast/entity.py +++ b/sdk/python/feast/entity.py @@ -173,13 +173,12 @@ def from_proto(cls, entity_proto: EntityProto): entity = cls( name=entity_proto.spec.name, join_keys=[entity_proto.spec.join_key], + value_type=ValueType(entity_proto.spec.value_type), description=entity_proto.spec.description, tags=dict(entity_proto.spec.tags), owner=entity_proto.spec.owner, ) - entity.value_type = ValueType(entity_proto.spec.value_type) - if entity_proto.meta.HasField("created_timestamp"): entity.created_timestamp = entity_proto.meta.created_timestamp.ToDatetime() if entity_proto.meta.HasField("last_updated_timestamp"):