fix

Author: xinrong-meng

python/pyspark/pandas/data_type_ops/num_ops.py

@@ -257,7 +257,7 @@ def truediv(left: PySparkColumn, right: Any) -> PySparkColumn:
                 ).otherwise(F.lit(np.inf).__div__(left))
             else:
                 return F.when(
-                    right == 0,
+                    F.lit(right == 0),
                     F.when(left < 0, F.lit(float("-inf")))
                     .when(left > 0, F.lit(float("inf")))
                     .otherwise(F.lit(np.nan)),
@@ -354,7 +354,7 @@ def truediv(left: PySparkColumn, right: Any) -> PySparkColumn:
                 )
             else:
                 return F.when(
-                    right == 0,
+                    F.lit(right == 0),
                     F.when(left < 0, F.lit(float("-inf")))
                     .when(left > 0, F.lit(float("inf")))
                     .otherwise(F.lit(np.nan)),