fix: raise error when not able to read path sparksource

ElliotNguyen68 · ElliotNguyen68 · commit 054a9765dd9d · 2024-03-12T16:31:42.000+07:00
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
@@ -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)
 
