From 126f350b5ab10dc28d09f9a3e5912c4f7a376345 Mon Sep 17 00:00:00 2001 From: tanlocnguyen Date: Tue, 12 Mar 2024 16:40:02 +0700 Subject: [PATCH] fix: raise error when not able to read path sparksource Signed-off-by: tanlocnguyen --- .../offline_stores/contrib/spark_offline_store/spark_source.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 1ff7e6de589..8cd392ce5df 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)