test

Author: Jing-song

src/azure-cli/azure/cli/command_modules/vm/_params.py

@@ -728,10 +728,6 @@ def load_arguments(self, _):
                    options_list=['--instance-ids', c.deprecate(target='--instance-id', redirect='--instance-ids', hide=True)])
         c.argument('force_update_os_disk_for_ephemeral', options_list=['--force-update-os-disk-for-ephemeral', '--update-os-disk'], arg_type=get_three_state_flag(), min_api='2024-03-01', help='Force update ephemeral OS disk for a virtual machine scale set VM.')
 
-    with self.argument_context('vmss scale', operation_group='virtual_machine_scale_sets') as c:
-        c.argument('extended_location', min_api='2020-12-01', help='The extended location of the Virtual Machine Scale Set.')
-
-
     with self.argument_context('vmss create', operation_group='virtual_machine_scale_sets') as c:
         VirtualMachineEvictionPolicyTypes = self.get_models('VirtualMachineEvictionPolicyTypes', resource_type=ResourceType.MGMT_COMPUTE)
 

src/azure-cli/azure/cli/command_modules/vm/custom.py

@@ -3925,21 +3925,17 @@ def restart_vmss(cmd, resource_group_name, vm_scale_set_name, instance_ids=None,
 
 
 # pylint: disable=inconsistent-return-statements
-def scale_vmss(cmd, resource_group_name, vm_scale_set_name, new_capacity, extended_location=None, no_wait=False):
+def scale_vmss(cmd, resource_group_name, vm_scale_set_name, new_capacity, no_wait=False):
     VirtualMachineScaleSet = cmd.get_models('VirtualMachineScaleSet')
     client = _compute_client_factory(cmd.cli_ctx)
     vmss = client.virtual_machine_scale_sets.get(resource_group_name, vm_scale_set_name)
     # pylint: disable=no-member
     if vmss.sku.capacity == new_capacity:
         return
 
-    if extended_location is not None:
-        ExtendedLocation = cmd.get_models('ExtendedLocation')
-        extended_location = ExtendedLocation(name=extended_location, type='EdgeZone')
-
     vmss.sku.capacity = new_capacity
-    vmss_new = VirtualMachineScaleSet(location=vmss.location, sku=vmss.sku) if extended_location is None \
-        else VirtualMachineScaleSet(location=vmss.location, sku=vmss.sku, extended_location=extended_location)
+    vmss_new = VirtualMachineScaleSet(location=vmss.location, sku=vmss.sku) if vmss.extended_location is None \
+        else VirtualMachineScaleSet(location=vmss.location, sku=vmss.sku, extended_location=vmss.extended_location)
     return sdk_no_wait(no_wait, client.virtual_machine_scale_sets.begin_create_or_update,
                        resource_group_name, vm_scale_set_name, vmss_new)