As of 2025-05-20, Set-AzContext worked for me for setting a subscription.
Set-AzContext -Subscription "xxxx-xxxx-xxxx-xxxx"