diff --git a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py index 1ff7e6de58..8cd392ce5d 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark_source.py @@ -8,7 +8,7 @@ from feast import flags_helper from feast.data_source import DataSource -from feast.errors import DataSourceNoNameException +from feast.errors import DataSourceNoNameException, DataSourceNotFoundException from feast.infra.offline_stores.offline_utils import get_temp_entity_table_name from feast.protos.feast.core.DataSource_pb2 import DataSource as DataSourceProto from feast.protos.feast.core.SavedDataset_pb2 import ( @@ -179,6 +179,7 @@ def get_table_query_string(self) -> str: logger.exception( "Spark read of file source failed.\n" + traceback.format_exc() ) + raise DataSourceNotFoundException(self.path) tmp_table_name = get_temp_entity_table_name() df.createOrReplaceTempView(tmp_table_name)