Merge pull request #75 from ExpediaGroup/feature/push_source_pydantic…

…_model

Feature/push source pydantic model

sdk/python/feast/expediagroup/pydantic_models/feature_view_model.py

@@ -6,7 +6,7 @@
 """
 import sys
 from datetime import datetime, timedelta
-from typing import Dict, List, Optional, Tuple
+from typing import Dict, List, Optional, Tuple, Union
 
 import dill
 from pydantic import BaseModel
@@ -15,6 +15,7 @@
 from feast.expediagroup.pydantic_models.data_source_model import (
     AnyBatchDataSource,
     KafkaSourceModel,
+    PushSourceModel,
     RequestSourceModel,
     SparkSourceModel,
 )
@@ -25,7 +26,7 @@
 from feast.on_demand_feature_view import OnDemandFeatureView
 
 # TO DO: Supported batch and supported streaming
-SUPPORTED_BATCH_DATA_SOURCES = [RequestSourceModel, SparkSourceModel]
+SUPPORTED_BATCH_DATA_SOURCES = [RequestSourceModel, SparkSourceModel, PushSourceModel]
 SUPPORTED_STREAM_DATA_SOURCES = [KafkaSourceModel]
 
 
@@ -63,7 +64,7 @@ class FeatureViewModel(BaseFeatureViewModel):
     original_entities: List[EntityModel] = []
     original_schema: Optional[List[FieldModel]]
     ttl: Optional[timedelta]
-    batch_source: AnyBatchDataSource
+    batch_source: Union[AnyBatchDataSource, PushSourceModel]
     stream_source: Optional[KafkaSourceModel]
     online: bool
     description: str