We don’t punish anything, nature punishes naturally everyone who are going against its rules, because life thrives on rules, not on disorder.
Soul to be free, needs cleaning, correcting itself, advancing, growing, doing favours for Gods, and fighting enemy.
By the time everything becomes fixed, and one is free again.
not advancing would be real punishment, because this way one will just dissipate into nothingness.
If one is on the path, everything can be corrected and cleaned.
if one did something horribly wrong, they would be punished if they don’t want to reward themselves with fruits of growth.
There is no punishment if one is willing to grow and help Gods.
If one is not willing to grow, then this is punishment by default and it is not because one did something wrong, because mistakes makes us learn and grow.
But because they are choose to not reward themselves in learning from their mistakes.