Whenever I check my water heater it is off. I will then light the pilot and turn the heater on. It will run properly until the end of the heat cycle, but once it is done it turns all the way off instead of just leaving the pilot on.

Replace thermocoupling.