How many miles ? The pump itself is outrageously expensive. If you are under 60k I would keep it. Over that, tough call. Over 90k just change it, its gonna break soon.
You can always take the car back for today, and return next week to complete the job. Don't wait too much tough, the thermostat is supposed to fail open, but that doesn't always work.