It should still handle PCIe probing and (logical) reconnection without a reboot, though, e.g.: PCIe redirection for a VM.
It handles warm reboots without power loss just fine, so it deciding now it needs to wait for power loss seems like a flaw.
If the SSD reacts to the start of a brown-out with supply voltage dropping way below spec as a signal that an unplanned power loss is happening, then it may do an emergency flush and shutdown that leaves it simply waiting for power to finish dropping to zero. It makes at least some sense for the drive to not try to wake up from that state without a clean power cycle.
I think "makes at least some sense" and "not particularly correct" can be true at the same time.