text: # pylint: disable=too-many-lines
  # coding=utf-8
  # --------------------------------------------------------------------------
  # Copyright (c) Microsoft Corporation. All rights reserved.
  # Licensed under the MIT License. See License.txt in the project root for license information.
  # Code generated by Microsoft (R) AutoRest Code Generator.
  # Changes may cause incorrect behavior and will be lost if the code is regenerated.
  # --------------------------------------------------------------------------
  from typing import Any, AsyncIterable, Callable, Dict, Optional, TypeVar, Union
  
  from azure.core.async_paging import AsyncItemPaged, AsyncList
  from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
  from azure.core.pipeline import PipelineResponse
  from azure.core.pipeline.transport import AsyncHttpResponse
  from azure.core.polling import AsyncLROPoller, AsyncNoPolling, AsyncPollingMethod
  from azure.core.rest import HttpRequest
  from azure.core.tracing.decorator import distributed_trace
  from azure.core.tracing.decorator_async import distributed_trace_async
  from azure.mgmt.core.exceptions import ARMErrorFormat
  from azure.mgmt.core.polling.async_arm_polling import AsyncARMPolling
  
  from ... import models as _models
  from ..._vendor import _convert_request
  from ...operations._virtual_machine_scale_sets_operations import build_create_or_update_request_initial, build_deallocate_request_initial, build_delete_instances_request_initial, build_delete_request_initial, build_force_recovery_service_fabric_platform_update_domain_walk_request, build_get_instance_view_request, build_get_os_upgrade_history_request, build_get_request, build_list_all_request, build_list_request, build_list_skus_request, build_perform_maintenance_request_initial, build_power_off_request_initial, build_redeploy_request_initial, build_reimage_all_request_initial, build_reimage_request_initial, build_restart_request_initial, build_start_request_initial, build_update_instances_request_initial, build_update_request_initial
  T = TypeVar('T')
  ClsType = Optional[Callable[[PipelineResponse[HttpRequest, AsyncHttpResponse], T, Dict[str, Any]], Any]]
  
  class VirtualMachineScaleSetsOperations:  # pylint: disable=too-many-public-methods
      """VirtualMachineScaleSetsOperations async operations.
  
      You should not instantiate this class directly. Instead, you should create a Client instance that
      instantiates it for you and attaches it as an attribute.
  
      :ivar models: Alias to model classes used in this operation group.
      :type models: ~azure.mgmt.compute.v2018_06_01.models
      :param client: Client for service requests.
      :param config: Configuration of service client.
      :param serializer: An object model serializer.
      :param deserializer: An object model deserializer.
      """
  
      models = _models
  
      def __init__(self, client, config, serializer, deserializer) -> None:
          self._client = client
          self._serialize = serializer
          self._deserialize = deserializer
          self._config = config
  
      async def _create_or_update_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          parameters: "_models.VirtualMachineScaleSet",
          **kwargs: Any
      ) -> "_models.VirtualMachineScaleSet":
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSet"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          _json = self._serialize.body(parameters, 'VirtualMachineScaleSet')
  
          request = build_create_or_update_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._create_or_update_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 201]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if response.status_code == 200:
              deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
  
          if response.status_code == 201:
              deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _create_or_update_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_create_or_update(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          parameters: "_models.VirtualMachineScaleSet",
          **kwargs: Any
      ) -> AsyncLROPoller["_models.VirtualMachineScaleSet"]:
          """Create or update a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set to create or update.
          :type vm_scale_set_name: str
          :param parameters: The scale set object.
          :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result
           of cls(response)
          :rtype:
           ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSet"]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._create_or_update_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  parameters=parameters,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              response = pipeline_response.http_response
              deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_create_or_update.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
      async def _update_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          parameters: "_models.VirtualMachineScaleSetUpdate",
          **kwargs: Any
      ) -> "_models.VirtualMachineScaleSet":
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSet"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          _json = self._serialize.body(parameters, 'VirtualMachineScaleSetUpdate')
  
          request = build_update_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._update_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _update_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_update(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          parameters: "_models.VirtualMachineScaleSetUpdate",
          **kwargs: Any
      ) -> AsyncLROPoller["_models.VirtualMachineScaleSet"]:
          """Update a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set to create or update.
          :type vm_scale_set_name: str
          :param parameters: The scale set object.
          :type parameters: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetUpdate
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either VirtualMachineScaleSet or the result
           of cls(response)
          :rtype:
           ~azure.core.polling.AsyncLROPoller[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSet"]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._update_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  parameters=parameters,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              response = pipeline_response.http_response
              deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_update.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
      async def _delete_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          
          request = build_delete_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              template_url=self._delete_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202, 204]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _delete_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_delete(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Deletes a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._delete_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  api_version=api_version,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_delete.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
      @distributed_trace_async
      async def get(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> "_models.VirtualMachineScaleSet":
          """Display information about a virtual machine scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: VirtualMachineScaleSet, or the result of cls(response)
          :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSet
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSet"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          
          request = build_get_request(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              template_url=self.get.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          deserialized = self._deserialize('VirtualMachineScaleSet', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      get.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}"}  # type: ignore
  
  
      async def _deallocate_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_deallocate_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._deallocate_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _deallocate_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_deallocate(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Deallocates specific virtual machines in a VM scale set. Shuts down the virtual machines and
          releases the compute resources. You are not billed for the compute resources that this virtual
          machine scale set deallocates.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._deallocate_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_deallocate.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate"}  # type: ignore
  
      async def _delete_instances_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceRequiredIDs')
  
          request = build_delete_instances_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._delete_instances_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _delete_instances_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_delete_instances(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Deletes virtual machines in a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._delete_instances_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_delete_instances.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete"}  # type: ignore
  
      @distributed_trace_async
      async def get_instance_view(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> "_models.VirtualMachineScaleSetInstanceView":
          """Gets the status of a VM scale set instance.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: VirtualMachineScaleSetInstanceView, or the result of cls(response)
          :rtype: ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetInstanceView
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSetInstanceView"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          
          request = build_get_instance_view_request(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              template_url=self.get_instance_view.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          deserialized = self._deserialize('VirtualMachineScaleSetInstanceView', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      get_instance_view.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView"}  # type: ignore
  
  
      @distributed_trace
      def list(
          self,
          resource_group_name: str,
          **kwargs: Any
      ) -> AsyncIterable["_models.VirtualMachineScaleSetListResult"]:
          """Gets a list of all VM scale sets under a resource group.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: An iterator like instance of either VirtualMachineScaleSetListResult or the result of
           cls(response)
          :rtype:
           ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetListResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSetListResult"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          def prepare_request(next_link=None):
              if not next_link:
                  
                  request = build_list_request(
                      resource_group_name=resource_group_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=self.list.metadata['url'],
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
  
              else:
                  
                  request = build_list_request(
                      resource_group_name=resource_group_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          async def extract_data(pipeline_response):
              deserialized = self._deserialize("VirtualMachineScaleSetListResult", pipeline_response)
              list_of_elem = deserialized.value
              if cls:
                  list_of_elem = cls(list_of_elem)
              return deserialized.next_link or None, AsyncList(list_of_elem)
  
          async def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
                  request,
                  stream=False,
                  **kwargs
              )
              response = pipeline_response.http_response
  
              if response.status_code not in [200]:
                  map_error(status_code=response.status_code, response=response, error_map=error_map)
                  raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
              return pipeline_response
  
  
          return AsyncItemPaged(
              get_next, extract_data
          )
      list.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets"}  # type: ignore
  
      @distributed_trace
      def list_all(
          self,
          **kwargs: Any
      ) -> AsyncIterable["_models.VirtualMachineScaleSetListWithLinkResult"]:
          """Gets a list of all VM Scale Sets in the subscription, regardless of the associated resource
          group. Use nextLink property in the response to get the next page of VM Scale Sets. Do this
          till nextLink is null to fetch all the VM Scale Sets.
  
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: An iterator like instance of either VirtualMachineScaleSetListWithLinkResult or the
           result of cls(response)
          :rtype:
           ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetListWithLinkResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSetListWithLinkResult"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          def prepare_request(next_link=None):
              if not next_link:
                  
                  request = build_list_all_request(
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=self.list_all.metadata['url'],
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
  
              else:
                  
                  request = build_list_all_request(
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          async def extract_data(pipeline_response):
              deserialized = self._deserialize("VirtualMachineScaleSetListWithLinkResult", pipeline_response)
              list_of_elem = deserialized.value
              if cls:
                  list_of_elem = cls(list_of_elem)
              return deserialized.next_link or None, AsyncList(list_of_elem)
  
          async def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
                  request,
                  stream=False,
                  **kwargs
              )
              response = pipeline_response.http_response
  
              if response.status_code not in [200]:
                  map_error(status_code=response.status_code, response=response, error_map=error_map)
                  raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
              return pipeline_response
  
  
          return AsyncItemPaged(
              get_next, extract_data
          )
      list_all.metadata = {'url': "/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets"}  # type: ignore
  
      @distributed_trace
      def list_skus(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> AsyncIterable["_models.VirtualMachineScaleSetListSkusResult"]:
          """Gets a list of SKUs available for your VM scale set, including the minimum and maximum VM
          instances allowed for each SKU.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: An iterator like instance of either VirtualMachineScaleSetListSkusResult or the result
           of cls(response)
          :rtype:
           ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetListSkusResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSetListSkusResult"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          def prepare_request(next_link=None):
              if not next_link:
                  
                  request = build_list_skus_request(
                      resource_group_name=resource_group_name,
                      vm_scale_set_name=vm_scale_set_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=self.list_skus.metadata['url'],
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
  
              else:
                  
                  request = build_list_skus_request(
                      resource_group_name=resource_group_name,
                      vm_scale_set_name=vm_scale_set_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          async def extract_data(pipeline_response):
              deserialized = self._deserialize("VirtualMachineScaleSetListSkusResult", pipeline_response)
              list_of_elem = deserialized.value
              if cls:
                  list_of_elem = cls(list_of_elem)
              return deserialized.next_link or None, AsyncList(list_of_elem)
  
          async def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
                  request,
                  stream=False,
                  **kwargs
              )
              response = pipeline_response.http_response
  
              if response.status_code not in [200]:
                  map_error(status_code=response.status_code, response=response, error_map=error_map)
                  raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
              return pipeline_response
  
  
          return AsyncItemPaged(
              get_next, extract_data
          )
      list_skus.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus"}  # type: ignore
  
      @distributed_trace
      def get_os_upgrade_history(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> AsyncIterable["_models.VirtualMachineScaleSetListOSUpgradeHistory"]:
          """Gets list of OS upgrades on a VM scale set instance.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: An iterator like instance of either VirtualMachineScaleSetListOSUpgradeHistory or the
           result of cls(response)
          :rtype:
           ~azure.core.async_paging.AsyncItemPaged[~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetListOSUpgradeHistory]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.VirtualMachineScaleSetListOSUpgradeHistory"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
          def prepare_request(next_link=None):
              if not next_link:
                  
                  request = build_get_os_upgrade_history_request(
                      resource_group_name=resource_group_name,
                      vm_scale_set_name=vm_scale_set_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=self.get_os_upgrade_history.metadata['url'],
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
  
              else:
                  
                  request = build_get_os_upgrade_history_request(
                      resource_group_name=resource_group_name,
                      vm_scale_set_name=vm_scale_set_name,
                      subscription_id=self._config.subscription_id,
                      api_version=api_version,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          async def extract_data(pipeline_response):
              deserialized = self._deserialize("VirtualMachineScaleSetListOSUpgradeHistory", pipeline_response)
              list_of_elem = deserialized.value
              if cls:
                  list_of_elem = cls(list_of_elem)
              return deserialized.next_link or None, AsyncList(list_of_elem)
  
          async def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
                  request,
                  stream=False,
                  **kwargs
              )
              response = pipeline_response.http_response
  
              if response.status_code not in [200]:
                  map_error(status_code=response.status_code, response=response, error_map=error_map)
                  raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
              return pipeline_response
  
  
          return AsyncItemPaged(
              get_next, extract_data
          )
      get_os_upgrade_history.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/osUpgradeHistory"}  # type: ignore
  
      async def _power_off_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_power_off_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._power_off_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _power_off_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_power_off(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Power off (stop) one or more virtual machines in a VM scale set. Note that resources are still
          attached and you are getting charged for the resources. Instead, use deallocate to release
          resources and avoid charges.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._power_off_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_power_off.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff"}  # type: ignore
  
      async def _restart_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_restart_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._restart_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _restart_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_restart(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Restarts one or more virtual machines in a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._restart_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_restart.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart"}  # type: ignore
  
      async def _start_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_start_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._start_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _start_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_start(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Starts one or more virtual machines in a VM scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._start_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_start.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start"}  # type: ignore
  
      async def _redeploy_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_redeploy_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._redeploy_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _redeploy_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_redeploy(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Shuts down all the virtual machines in the virtual machine scale set, moves them to a new node,
          and powers them back on.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._redeploy_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_redeploy.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/redeploy"}  # type: ignore
  
      async def _perform_maintenance_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_perform_maintenance_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._perform_maintenance_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _perform_maintenance_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_perform_maintenance(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Perform maintenance on one or more virtual machines in a VM scale set. Operation on instances
          which are not eligible for perform maintenance will be failed. Please refer to best practices
          for more details:
          https://docs.microsoft.com/en-us/azure/virtual-machine-scale-sets/virtual-machine-scale-sets-maintenance-notifications.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._perform_maintenance_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_perform_maintenance.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/performMaintenance"}  # type: ignore
  
      async def _update_instances_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceRequiredIDs')
  
          request = build_update_instances_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._update_instances_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _update_instances_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_update_instances(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Upgrades one or more virtual machines to the latest SKU set in the VM scale set model.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceRequiredIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._update_instances_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_update_instances.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade"}  # type: ignore
  
      async def _reimage_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_scale_set_reimage_input: Optional["_models.VirtualMachineScaleSetReimageParameters"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_scale_set_reimage_input is not None:
              _json = self._serialize.body(vm_scale_set_reimage_input, 'VirtualMachineScaleSetReimageParameters')
          else:
              _json = None
  
          request = build_reimage_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._reimage_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _reimage_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_reimage(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_scale_set_reimage_input: Optional["_models.VirtualMachineScaleSetReimageParameters"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Reimages (upgrade the operating system) one or more virtual machines in a VM scale set which
          don't have a ephemeral OS disk, for virtual machines who have a ephemeral OS disk the virtual
          machine is reset to initial state.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_scale_set_reimage_input: Parameters for Reimaging VM ScaleSet. Default value is None.
          :type vm_scale_set_reimage_input:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetReimageParameters
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._reimage_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_scale_set_reimage_input=vm_scale_set_reimage_input,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_reimage.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage"}  # type: ignore
  
      async def _reimage_all_initial(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> None:
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
  
          if vm_instance_i_ds is not None:
              _json = self._serialize.body(vm_instance_i_ds, 'VirtualMachineScaleSetVMInstanceIDs')
          else:
              _json = None
  
          request = build_reimage_all_request_initial(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              content_type=content_type,
              json=_json,
              template_url=self._reimage_all_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200, 202]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          if cls:
              return cls(pipeline_response, None, {})
  
      _reimage_all_initial.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall"}  # type: ignore
  
  
      @distributed_trace_async
      async def begin_reimage_all(  # pylint: disable=inconsistent-return-statements
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> AsyncLROPoller[None]:
          """Reimages all the disks ( including data disks ) in the virtual machines in a VM scale set. This
          operation is only supported for managed disks.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param vm_instance_i_ds: A list of virtual machine instance IDs from the VM scale set. Default
           value is None.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2018_06_01.models.VirtualMachineScaleSetVMInstanceIDs
          :keyword callable cls: A custom type or function that will be passed the direct response
          :keyword str continuation_token: A continuation token to restart a poller from a saved state.
          :keyword polling: By default, your polling method will be AsyncARMPolling. Pass in False for
           this operation to not poll, or pass in your own initialized polling object for a personal
           polling strategy.
          :paramtype polling: bool or ~azure.core.polling.AsyncPollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of AsyncLROPoller that returns either None or the result of cls(response)
          :rtype: ~azure.core.polling.AsyncLROPoller[None]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, AsyncPollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType[None]
          lro_delay = kwargs.pop(
              'polling_interval',
              self._config.polling_interval
          )
          cont_token = kwargs.pop('continuation_token', None)  # type: Optional[str]
          if cont_token is None:
              raw_result = await self._reimage_all_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  vm_instance_i_ds=vm_instance_i_ds,
                  api_version=api_version,
                  content_type=content_type,
                  cls=lambda x,y,z: x,
                  **kwargs
              )
          kwargs.pop('error_map', None)
  
          def get_long_running_output(pipeline_response):
              if cls:
                  return cls(pipeline_response, None, {})
  
  
          if polling is True: polling_method = AsyncARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = AsyncNoPolling()
          else: polling_method = polling
          if cont_token:
              return AsyncLROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          return AsyncLROPoller(self._client, raw_result, get_long_running_output, polling_method)
  
      begin_reimage_all.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall"}  # type: ignore
  
      @distributed_trace_async
      async def force_recovery_service_fabric_platform_update_domain_walk(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          platform_update_domain: int,
          **kwargs: Any
      ) -> "_models.RecoveryWalkResponse":
          """Manual platform update domain walk to update virtual machines in a service fabric virtual
          machine scale set.
  
          :param resource_group_name: The name of the resource group.
          :type resource_group_name: str
          :param vm_scale_set_name: The name of the VM scale set.
          :type vm_scale_set_name: str
          :param platform_update_domain: The platform update domain for which a manual recovery walk is
           requested.
          :type platform_update_domain: int
          :keyword callable cls: A custom type or function that will be passed the direct response
          :return: RecoveryWalkResponse, or the result of cls(response)
          :rtype: ~azure.mgmt.compute.v2018_06_01.models.RecoveryWalkResponse
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.RecoveryWalkResponse"]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          api_version = kwargs.pop('api_version', "2018-06-01")  # type: str
  
          
          request = build_force_recovery_service_fabric_platform_update_domain_walk_request(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              api_version=api_version,
              platform_update_domain=platform_update_domain,
              template_url=self.force_recovery_service_fabric_platform_update_domain_walk.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = await self._client._pipeline.run(  # pylint: disable=protected-access
              request,
              stream=False,
              **kwargs
          )
          response = pipeline_response.http_response
  
          if response.status_code not in [200]:
              map_error(status_code=response.status_code, response=response, error_map=error_map)
              raise HttpResponseError(response=response, error_format=ARMErrorFormat)
  
          deserialized = self._deserialize('RecoveryWalkResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      force_recovery_service_fabric_platform_update_domain_walk.metadata = {'url': "/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/forceRecoveryServiceFabricPlatformUpdateDomainWalk"}  # type: ignore
  
  