refactor(datafusion): get name from expr

ibis-project · Oct 5, 2022 · fea3e5b · fea3e5b
1 parent 345b3b8
commit fea3e5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ibis/backends/datafusion/__init__.py b/ibis/backends/datafusion/__init__.py
@@ -215,7 +215,7 @@ def execute(
             return output.to_pandas()
         elif isinstance(expr, ir.Column):
             series = output.to_pandas()
-            series.name = "tmp"
+            series.name = expr.get_name()
             return series
         elif isinstance(expr, ir.Scalar):
             return output.as_py()

diff --git a/ibis/backends/datafusion/tests/test_udf.py b/ibis/backends/datafusion/tests/test_udf.py
@@ -37,7 +37,7 @@ def test_udf(alltypes):
 
 
 def test_multiple_argument_udf(alltypes):
-    expr = my_add(alltypes.smallint_col, alltypes.int_col)
+    expr = my_add(alltypes.smallint_col, alltypes.int_col).name("tmp")
     result = expr.execute()
 
     df = alltypes[['smallint_col', 'int_col']].execute()