Skip to content

Commit 485b78c

Browse files
committed
fix test
1 parent 51ae36a commit 485b78c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/pyspark/sql/tests/pandas/test_pandas_grouped_map.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,8 @@ def check_register_grouped_map_udf(self):
222222
errorClass="INVALID_UDF_EVAL_TYPE",
223223
messageParameters={
224224
"eval_type": "SQL_BATCHED_UDF, SQL_ARROW_BATCHED_UDF, SQL_SCALAR_PANDAS_UDF, "
225-
"SQL_SCALAR_PANDAS_ITER_UDF or SQL_GROUPED_AGG_PANDAS_UDF"
225+
"SQL_SCALAR_PANDAS_ITER_UDF, SQL_GROUPED_AGG_PANDAS_UDF or "
226+
"SQL_GROUPED_AGG_ARROW_UDF"
226227
},
227228
)
228229

0 commit comments

Comments
 (0)