feat: Set value_type of entity directly in from_proto (#5092)

Set value_type of entity directly in from_proto Signed-off-by: hkuepers <[email protected]> Co-authored-by: hkuepers <[email protected]>
feast-dev · Feb 25, 2025 · 90e7498 · 90e7498
1 parent d1bf58f
commit 90e7498
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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"):