Spark exception error using pandas_udf with logical statement

Question

I am trying to deploy a simple if-else function specifically using pandas_udf. Here is the code:

from pyspark.sql import *
from pyspark.sql.functions import *