pyspark agg() with groupby() is not working using pandas_udf

Question

When I am using one of min, sum, max, count, mean or std the pyspark groupby().agg() is working fine. But if I provide any of median_udf, range_udf or quantile_udf user-defined