fix

logicalclocks · Dec 20, 2024 · 6a96b06 · 6a96b06
1 parent 22162b2
commit 6a96b06
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/hsfs/engine/spark.py b/python/hsfs/engine/spark.py
@@ -523,7 +523,7 @@ def save_stream_dataframe(
                 "wait_for_online_ingestion", False
             ):
                 feature_group.get_latest_online_ingestion().wait_for_completion(
-                    options=offline_write_options.get("online_ingestion_options", {})
+                    options=write_options.get("online_ingestion_options", {})
                 )
 
         return query
@@ -586,7 +586,7 @@ def _save_online_dataframe(self, feature_group, dataframe, write_options):
             "wait_for_online_ingestion", False
         ):
             feature_group.get_latest_online_ingestion().wait_for_completion(
-                options=offline_write_options.get("online_ingestion_options", {})
+                options=write_options.get("online_ingestion_options", {})
             )
 
     def _get_headers(