@f.pandas_udf(returnType= DoubleType()) def square(r : pd.Series) -> pd.Series: print(\'In pandas Udf square\') offset_value = offset.value return (r