If God had left Noah to go through the judgment, He would have left him on the earth, not floating above it, safe and sound, in the ark.
Those not in the ark were taken out - to judgement. Noah was not removed from the Earth. He never left the Earth. The flood waters were still a part of the Earth.