Revert "[SPARK-54943][PYTHON][TESTS][FOLLOW-UP] Mistake Commit"

revert a mistake commit

Closes #54050 from zhengruifeng/revert_mistake.

Authored-by: Ruifeng Zheng <ruifengz@apache.org>
Signed-off-by: Ruifeng Zheng <ruifengz@apache.org>