I'd say get your TDI back if you can.
I turned in a near perfect 22k miles 2013 Jetta TDI Premium in late 2016, wish I had asked about getting it back. I bought a 2013 Golf TDI DSG equipped on Dec. 27th 2018 with 26k on the clock. Got a great price on it for being a return customer. I'm beginning to think I bought a load of troubles now.
Some times low mileage doesn't mean much! As of today I've been driving a loaner, a Jetta TSi/DSG, not bad but not my Golf. since Jan. 14th, so the dealer has had my low miles Golf as long as I have. The CEL light came on shortly after heading home from a trip to the dealer so I could put my winter tires on. One of the 100 or so items of the CPO checklist is to make sure that no locking lug bolts are left on the car. They forgot. Which leaves me wondering what else they might have forgotten. Anyway, the dealer took off the locking lug bolts and installed my winter tires, free of charge. Nice. About half way home the CEL came on. I go back to the dealer a few days later so they could read the codes, turned out to be bad news. The EGR was part of that CEL, the other being the DPF which seems to be clogged. So I get new EGR and DPF stuff, and probably a few sensors replaced. But the worst part is it has a bad/bent axle. WHF? And that is why my Golf is still at the dealer, not one replacement axle to be found in the USA! What? I'm hoping they'll get the repairs done soon, I'm tired of driving their billboard car.
At any rate, if your Jetta/Golf has a funny hopping feeling to it in quick (not racing) cornering under acceleration, or even straight line under hard, but not floored, acceleration, check the axles. I did notice on the 20 minute test drive that it seemed kind of weird to have a hopping sensation, but I didn't really notice it until a week or so later, and it seemed to be getting worse. Par for the course here, when the service tech test drove the car he didn't find anything wrong. So I took him along for another test drive, and he then understood what the car was doing. A quick check by a mechanic confirmed there was a problem, a bad axle. So they'll be installing new CV joints too.
Wish I had my old Jetta back, someone got a very nice car.