add return case of entity is str

Signed-off-by: tanlocnguyen <tanlocnguyen296@gmail.com>
feast-dev · Mar 6, 2024 · 0aef8d9 · 0aef8d9
1 parent 1ef7de9
commit 0aef8d9
Showing 1 changed file with 1 addition and 0 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
@@ -529,6 +529,7 @@ def _upload_entity_df(
         return
     elif isinstance(entity_df, str):
         spark_session.sql(entity_df).createOrReplaceTempView(table_name)
+        return 
     elif isinstance(entity_df, pyspark.sql.DataFrame):
         entity_df.createOrReplaceTempView(table_name)
         return