Set value_type of entity directly in from_proto

hkuepers · hkuepers · commit fb988ed74ff2 · 2025-02-25T17:27:02.000+01:00
Signed-off-by: hkuepers &lt;hanno.kuepers@ratepay.com&gt;
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"):
