fix: index out of range error in feast integration

probably-nothing-labs · Dec 9, 2024 · 72e4525 · 72e4525
1 parent d5a056c
commit 72e4525
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/py-denormalized/python/denormalized/feast_data_stream.py b/py-denormalized/python/denormalized/feast_data_stream.py
@@ -108,10 +108,11 @@ def write_feast_feature(
         """
 
         def _sink_to_feast(rb: pa.RecordBatch):
-            df = rb.to_pandas()
-            try:
-                feature_store.push(source_name, df, to=PushMode.ONLINE)
-            except Exception as e:
-                print(e)
+            if len(rb):
+                df = rb.to_pandas()
+                try:
+                    feature_store.push(source_name, df, to=PushMode.ONLINE)
+                except Exception as e:
+                    print(e)
 
         self.ds.sink_python(_sink_to_feast)