SiddheshDesai is right, you need automatic upgrades ... but you no longer need to create a new Uniform scale set to use it :) you can now use upgrade_mode = "Automatic" on Flex scale sets!
https://techcommunity.microsoft.com/blog/azurecompute/general-availability-upgrade-policies-on-virtual-machine-scale-sets-with-flexibl/4287334