(fix) EntitySQLEmptyResults lint error

feast-dev · Mar 6, 2024 · 388413c · 388413c
1 parent ceda87f
commit 388413c
Showing 1 changed file with 3 additions and 5 deletions.
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 = (