[SPARK-3781] code Style format

I think this is equivalent but I'm not 100% sure.