To play it safe in terms of spec, buy a used Apple drive.
To save money and get more speed: It may depend a bit on the exact model of MacBook Pro. If it has a discrete GPU then heat will be more of an issue. If it's an Intel iGPU only, things are much more relaxed.
It's not so important to stick to the specific brand (which I would recommend otherwise most of the time).
Most issues with replacement SSDs for this model come to down to connection problems. With a drive from that manufacturer/vendor, I'd say exchanging a faulty drive or a simple reseating of perfectly fine module should solve these problems.
But the more flexible and cost effective way is going for an established seller and manufacturer of M2-adapters. There are quite some a bit too cheap offerings out there.
Then look for aa matching SSD chewing gum strip. That is a bit tough for dGPU MBPs. Some drives do drain quite a bit of power and do get hotter than others.
Looking for your personal sweet spot might require some serious spec digging. I used a Samsung 970evo in this configuration on a few machines, and it does drain too much power (tolerable) and it can get a bit too warm (again: tolerable), in an Intel iGPU only MBP 2015.
Comparing specs and sensor readings show significant differences between the stock drives and the replacements. In real life this was negligible in terms of fan speed or battery life. This would not be the case if you'd power another external drive, which would draw quite a bit more pwer from the battery.
If it's still under warranty/AppleCare, keep in mind that you need the original drive (re-installed in need) should you want from Apple to service such a unit.