Add KafkaSourceModel to AnyDataSource

ExpediaGroup · Dec 1, 2023 · 331c870 · 331c870
1 parent f15a03c
commit 331c870
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/python/feast/expediagroup/pydantic_models/data_source_model.py b/sdk/python/feast/expediagroup/pydantic_models/data_source_model.py
@@ -278,6 +278,6 @@ def from_data_source(
 # https://blog.devgenius.io/deserialize-child-classes-with-pydantic-that-gonna-work-784230e1cf83
 # This lets us discriminate child classes of DataSourceModel with type hints.
 AnyDataSource = Annotated[
-    Union[RequestSourceModel, SparkSourceModel],
+    Union[RequestSourceModel, SparkSourceModel, KafkaSourceModel],
     PydanticField(discriminator="model_type"),
 ]