fix(datafusion): bring back UDF registration

ibis-project · Nov 20, 2023 · 43084fa · 43084fa
1 parent 669b352
commit 43084fa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ibis/backends/datafusion/__init__.py b/ibis/backends/datafusion/__init__.py
@@ -474,6 +474,7 @@ def to_pyarrow_batches(
     ) -> pa.ipc.RecordBatchReader:
         pa = self._import_pyarrow()
 
+        self._register_udfs(expr)
         self._register_in_memory_tables(expr)
 
         sql = self.compile(expr.as_table(), params=params, **kwargs)