added push source as batch source for feature view

ExpediaGroup · Jan 16, 2024 · 341d77e · 341d77e
1 parent a51e70a
commit 341d77e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/expediagroup/pydantic_models/feature_view_model.py b/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
@@ -64,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