Pandas UDF and Python Type Hint in Apache Spark 3