fix: Wrap LocalOutputNode return value in ArrowTableValue for consist… (#6286)

cutoutsy · web-flow · commit a16cd5509835 · 2026-04-16T19:53:30.000+05:30
* fix: Wrap LocalOutputNode return value in ArrowTableValue for consistent DAG output

Signed-off-by: cutoutsy &lt;cutoutsy@gmail.com&gt;

* fix: Update test to access ArrowTableValue.data.num_rows instead of num_rows

Signed-off-by: cutoutsy &lt;cutoutsy@gmail.com&gt;

---------

Signed-off-by: cutoutsy &lt;cutoutsy@gmail.com&gt;
diff --git a/sdk/python/feast/infra/compute_engines/local/nodes.py b/sdk/python/feast/infra/compute_engines/local/nodes.py
@@ -361,10 +361,11 @@ def __init__(
 
     def execute(self, context: ExecutionContext) -> ArrowTableValue:
         input_table = self.get_single_table(context).data
-        context.node_outputs[self.name] = input_table
+        output = ArrowTableValue(data=input_table)
+        context.node_outputs[self.name] = output
 
         if input_table.num_rows == 0:
-            return input_table
+            return output
 
         if self.feature_view.online:
             online_store = context.online_store
@@ -403,4 +404,4 @@ def execute(self, context: ExecutionContext) -> ArrowTableValue:
                 progress=lambda x: None,
             )
 
-        return input_table
+        return output
diff --git a/sdk/python/tests/unit/infra/compute_engines/local/test_nodes.py b/sdk/python/tests/unit/infra/compute_engines/local/test_nodes.py
@@ -216,7 +216,8 @@ def test_local_output_node():
     node.add_input(MagicMock())
     node.inputs[0].name = "source"
     result = node.execute(context)
-    assert result.num_rows == 4
+    assert isinstance(result, ArrowTableValue)
+    assert result.data.num_rows == 4
 
 
 def test_local_output_node_online_write_default_batch():
