updated FeatureViewProjection batch_source serialization

franciscojavierarceo · franciscojavierarceo · commit 34f99b9dc684 · 2024-09-21T11:31:26.000-04:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/feast/feature_view_projection.py b/sdk/python/feast/feature_view_projection.py
@@ -73,7 +73,12 @@ def to_proto(self) -> FeatureViewProjectionProto:
         return feature_reference_proto
 
     @staticmethod
-    def from_proto(proto: FeatureViewProjectionProto):
+    def from_proto(proto: FeatureViewProjectionProto) -> "FeatureViewProjection":
+        batch_source = (
+            DataSource.from_proto(proto.batch_source)
+            if str(getattr(proto, "batch_source"))
+            else None
+        )
         feature_view_projection = FeatureViewProjection(
             name=proto.feature_view_name,
             name_alias=proto.feature_view_name_alias or None,
@@ -83,7 +88,7 @@ def from_proto(proto: FeatureViewProjectionProto):
             timestamp_field=proto.timestamp_field or None,
             date_partition_column=proto.date_partition_column or None,
             created_timestamp_column=proto.created_timestamp_column or None,
-            batch_source=proto.batch_source or None,
+            batch_source=batch_source,
         )
         for feature_column in proto.feature_columns:
             feature_view_projection.features.append(Field.from_proto(feature_column))
