From 331c8708806c037250ed512515a2bb4ba7edfdf9 Mon Sep 17 00:00:00 2001 From: Matt Carlin Date: Fri, 1 Dec 2023 09:28:54 -0600 Subject: [PATCH] Add KafkaSourceModel to AnyDataSource --- .../feast/expediagroup/pydantic_models/data_source_model.py | 2 +- 1 file changed, 1 insertion(+), 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 index f6bc027818..6aece6025b 100644 --- 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"), ]