Thinkpad x1 Gen 9 SSD replacement

I'm planning to upgrade the SSD in my Thinkpad X1 Gen 9 to a WD Black SN850X 4TB. I've heard great things about this drive's performance, but I'm a bit concerned about its power and heat usage. Has anyone else out there done this upgrade and noticed any significant differences in battery life or thermal performance?