diff --git a/python/pyarrow/_compute.pyx b/python/pyarrow/_compute.pyx index 607f6723da3fd..cec849a01f585 100644 --- a/python/pyarrow/_compute.pyx +++ b/python/pyarrow/_compute.pyx @@ -2861,9 +2861,9 @@ def register_vector_function(func, function_name, function_doc, in_types, out_ty >>> answer [ - 0.3333333333333333, - 0.6666666666666666, - 1 + 0.3333333333333333, + 0.6666666666666666, + 1 ] """ return _register_user_defined_function(get_register_vector_function(), diff --git a/python/pyarrow/tests/test_udf.py b/python/pyarrow/tests/test_udf.py index 43e7f9d0ce3ab..62d1eb5bafd4f 100644 --- a/python/pyarrow/tests/test_udf.py +++ b/python/pyarrow/tests/test_udf.py @@ -841,7 +841,7 @@ def test_hash_agg_random(sum_agg_func_fixture): def test_vector_basic(unary_vector_func_fixture): arr = pa.array([10.0, 20.0, 30.0, 40.0, 50.0], pa.float64()) result = pc.call_function("y=pct_rank(x)", [arr]) - expected = unary_vector_func_fixture(None, arr) + expected = unary_vector_func_fixture[0](None, arr) assert result == expected @@ -849,7 +849,7 @@ def test_vector_basic(unary_vector_func_fixture): def test_vector_empty(unary_vector_func_fixture): arr = pa.array([1], pa.float64()) result = pc.call_function("y=pct_rank(x)", [arr]) - expected = unary_vector_func_fixture(None, arr) + expected = unary_vector_func_fixture[0](None, arr) assert result == expected