Something very wrong with Snowflake Dataframe. I see the same results.
Based on the query history snowflake is running below SQL for result_df = some_df.with_column('the_sum', call_udf('test.public.some_sum', some_df['A'], some_df['B']))
operation
SELECT "A", "B", test.public.some_sum("A", "B") AS "THE_SUM" FROM "TEST"."PUBLIC"."SNOWPARK_TEMP_TABLE_54NO971VY9" LIMIT 10;
I tried running the SQL UDF call instead of dataframe UDF call. It works fine