Set value_type of entity directly in from_proto

Signed-off-by: hkuepers <[email protected]>
feast-dev · Feb 25, 2025 · 632312c · 632312c
1 parent e25ca4f
commit 632312c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sdk/python/feast/entity.py b/sdk/python/feast/entity.py
@@ -172,14 +172,13 @@ def from_proto(cls, entity_proto: EntityProto):
         """
         entity = cls(
             name=entity_proto.spec.name,
+            valuetType=ValueType(entity_proto.spec.value_type),
             join_keys=[entity_proto.spec.join_key],
             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"):