Why does decreasing partition count prevent a StageFailure due to large size of serialized results? I am trying to generate a randomly sorted dataframe on databricks.