From 349655dbb6f6438e57ed5b8e898566221fb0e194 Mon Sep 17 00:00:00 2001 From: aurobindoc Date: Tue, 15 Mar 2022 13:18:37 +0530 Subject: [PATCH] create SparkSource inside try catch Signed-off-by: aurobindoc --- sdk/python/feast/data_source.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/sdk/python/feast/data_source.py b/sdk/python/feast/data_source.py index 15ce0c23773..938ac734dcb 100644 --- a/sdk/python/feast/data_source.py +++ b/sdk/python/feast/data_source.py @@ -249,7 +249,14 @@ def from_proto(data_source: DataSourceProto) -> Any: ): data_source_obj = KinesisSource.from_proto(data_source) else: - raise ValueError("Could not identify the source type being added.") + try: + from feast.infra.offline_stores.contrib.spark_offline_store.spark_source import ( + SparkSource, + ) + + data_source_obj = SparkSource.from_proto(data_source) + except Exception: + raise ValueError("Could not identify the source type being added.") return data_source_obj