No, we do not see Pucey in the Final Battle of Hogwarts, so it's safe to presume that he didn't come back to Hogwarts for it.