fix udtf

Signed-off-by: Runji Wang <[email protected]>
risingwavelabs · May 7, 2024 · 3c6e87a · 3c6e87a
1 parent 5341ee4
commit 3c6e87a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/expr/core/src/table_function/user_defined.rs b/src/expr/core/src/table_function/user_defined.rs
@@ -110,7 +110,8 @@ impl UserDefinedTableFunction {
         // compact the input chunk and record the row mapping
         let visible_rows = direct_input.visibility().iter_ones().collect::<Vec<_>>();
         // this will drop invisible rows
-        let arrow_input = UdfArrowConvert.to_record_batch(self.arg_schema.clone(), input)?;
+        let arrow_input =
+            UdfArrowConvert.to_record_batch(self.arg_schema.clone(), &direct_input)?;
 
         // call UDTF
         #[for_await]