(fix) EntitySQLEmptyResults lint error

ElliotNguyen68 · ElliotNguyen68 · commit 388413cd122e · 2024-03-06T16:46:35.000+07:00
diff --git a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark.py b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/spark.py
@@ -478,13 +478,11 @@ def _get_entity_df_event_timestamp_range(
         # from table
         if isinstance(entity_df, str): 
             df = spark_session.sql(entity_df).select(entity_df_event_timestamp_col)
+            # Checks if executing entity sql resulted in any data
+            if df.rdd.isEmpty():
+                raise EntitySQLEmptyResults(entity_df)
         else:
             df = entity_df
-
-        # Checks if executing entity sql resulted in any data
-        if df.rdd.isEmpty():
-            raise EntitySQLEmptyResults(entity_df)
-
         # TODO(kzhang132): need utc conversion here.
 
         entity_df_event_timestamp_range = (
