text: # 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.
  # --------------------------------------------------------------------------
  import functools
  from typing import Any, Callable, Dict, Generic, Iterable, Optional, TypeVar, Union
  import warnings
  
  from azure.core.exceptions import ClientAuthenticationError, HttpResponseError, ResourceExistsError, ResourceNotFoundError, map_error
  from azure.core.paging import ItemPaged
  from azure.core.pipeline import PipelineResponse
  from azure.core.pipeline.transport import HttpResponse
  from azure.core.polling import LROPoller, NoPolling, PollingMethod
  from azure.core.rest import HttpRequest
  from azure.core.tracing.decorator import distributed_trace
  from azure.mgmt.core.exceptions import ARMErrorFormat
  from azure.mgmt.core.polling.arm_polling import ARMPolling
  from msrest import Serializer
  
  from .. import models as _models
  from .._vendor import _convert_request, _format_url_section
  T = TypeVar('T')
  JSONType = Any
  ClsType = Optional[Callable[[PipelineResponse[HttpRequest, HttpResponse], T, Dict[str, Any]], Any]]
  
  _SERIALIZER = Serializer()
  _SERIALIZER.client_side_validation = False
  
  def build_create_or_update_request_initial(
      resource_group_name: str,
      name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("name", name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="PUT",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_delete_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="DELETE",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_get_request(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="GET",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_deallocate_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_delete_instances_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_get_instance_view_request(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/instanceView')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="GET",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_list_request(
      resource_group_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="GET",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_list_all_request(
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/providers/Microsoft.Compute/virtualMachineScaleSets')
      path_format_arguments = {
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="GET",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_list_skus_request(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="GET",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_power_off_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_restart_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_start_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_update_instances_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      *,
      json: JSONType = None,
      content: Any = None,
      **kwargs: Any
  ) -> HttpRequest:
      content_type = kwargs.pop('content_type', None)  # type: Optional[str]
  
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      if content_type is not None:
          header_parameters['Content-Type'] = _SERIALIZER.header("content_type", content_type, 'str')
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          json=json,
          content=content,
          **kwargs
      )
  
  
  def build_reimage_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  
  def build_reimage_all_request_initial(
      resource_group_name: str,
      vm_scale_set_name: str,
      subscription_id: str,
      **kwargs: Any
  ) -> HttpRequest:
      api_version = "2016-04-30-preview"
      accept = "application/json"
      # Construct URL
      url = kwargs.pop("template_url", '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall')
      path_format_arguments = {
          "resourceGroupName": _SERIALIZER.url("resource_group_name", resource_group_name, 'str'),
          "vmScaleSetName": _SERIALIZER.url("vm_scale_set_name", vm_scale_set_name, 'str'),
          "subscriptionId": _SERIALIZER.url("subscription_id", subscription_id, 'str'),
      }
  
      url = _format_url_section(url, **path_format_arguments)
  
      # Construct parameters
      query_parameters = kwargs.pop("params", {})  # type: Dict[str, Any]
      query_parameters['api-version'] = _SERIALIZER.query("api_version", api_version, 'str')
  
      # Construct headers
      header_parameters = kwargs.pop("headers", {})  # type: Dict[str, Any]
      header_parameters['Accept'] = _SERIALIZER.header("accept", accept, 'str')
  
      return HttpRequest(
          method="POST",
          url=url,
          params=query_parameters,
          headers=header_parameters,
          **kwargs
      )
  
  class VirtualMachineScaleSetsOperations(object):
      """VirtualMachineScaleSetsOperations 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.v2016_04_30_preview.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):
          self._client = client
          self._serialize = serializer
          self._deserialize = deserializer
          self._config = config
  
      def _create_or_update_initial(
          self,
          resource_group_name: str,
          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', {}))
  
          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,
              name=name,
              subscription_id=self._config.subscription_id,
              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 = self._client._pipeline.run(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
      def begin_create_or_update(
          self,
          resource_group_name: str,
          name: str,
          parameters: "_models.VirtualMachineScaleSet",
          **kwargs: Any
      ) -> LROPoller["_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 name: The name of the VM scale set to create or update.
          :type name: str
          :param parameters: The scale set object.
          :type parameters: ~azure.mgmt.compute.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either VirtualMachineScaleSet or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSet]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          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 = self._create_or_update_initial(
                  resource_group_name=resource_group_name,
                  name=name,
                  parameters=parameters,
                  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 = ARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _delete_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          
          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,
              template_url=self._delete_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _delete_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}'}  # type: ignore
  
  
      @distributed_trace
      def begin_delete(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = self._delete_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
      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.v2016_04_30_preview.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', {}))
  
          
          request = build_get_request(
              resource_group_name=resource_group_name,
              vm_scale_set_name=vm_scale_set_name,
              subscription_id=self._config.subscription_id,
              template_url=self.get.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = self._client._pipeline.run(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
  
  
      def _deallocate_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _deallocate_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/deallocate'}  # type: ignore
  
  
      @distributed_trace
      def begin_deallocate(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _delete_instances_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _delete_instances_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/delete'}  # type: ignore
  
  
      @distributed_trace
      def begin_delete_instances(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
      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.v2016_04_30_preview.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', {}))
  
          
          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,
              template_url=self.get_instance_view.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = self._client._pipeline.run(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
      ) -> Iterable["_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.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetListResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          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,
                      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,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          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, iter(list_of_elem)
  
          def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = self._client._pipeline.run(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 ItemPaged(
              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
      ) -> Iterable["_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.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetListWithLinkResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          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,
                      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,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          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, iter(list_of_elem)
  
          def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = self._client._pipeline.run(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 ItemPaged(
              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
      ) -> Iterable["_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.paging.ItemPaged[~azure.mgmt.compute.v2016_04_30_preview.models.VirtualMachineScaleSetListSkusResult]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          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,
                      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,
                      template_url=next_link,
                  )
                  request = _convert_request(request)
                  request.url = self._client.format_url(request.url)
                  request.method = "GET"
              return request
  
          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, iter(list_of_elem)
  
          def get_next(next_link=None):
              request = prepare_request(next_link)
  
              pipeline_response = self._client._pipeline.run(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 ItemPaged(
              get_next, extract_data
          )
      list_skus.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/skus'}  # type: ignore
  
      def _power_off_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _power_off_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/poweroff'}  # type: ignore
  
  
      @distributed_trace
      def begin_power_off(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _restart_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _restart_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/restart'}  # type: ignore
  
  
      @distributed_trace
      def begin_restart(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _start_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _start_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/start'}  # type: ignore
  
  
      @distributed_trace
      def begin_start(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: Optional["_models.VirtualMachineScaleSetVMInstanceIDs"] = None,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.
          :type vm_instance_i_ds:
           ~azure.mgmt.compute.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _update_instances_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          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,
              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 = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _update_instances_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/manualupgrade'}  # type: ignore
  
  
      @distributed_trace
      def begin_update_instances(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          vm_instance_i_ds: "_models.VirtualMachineScaleSetVMInstanceRequiredIDs",
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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.v2016_04_30_preview.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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          content_type = kwargs.pop('content_type', "application/json")  # type: Optional[str]
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = 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,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _reimage_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          
          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,
              template_url=self._reimage_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _reimage_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimage'}  # type: ignore
  
  
      @distributed_trace
      def begin_reimage(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """Reimages (upgrade the operating system) 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
          :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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = self._reimage_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  
      def _reimage_all_initial(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> Optional["_models.OperationStatusResponse"]:
          cls = kwargs.pop('cls', None)  # type: ClsType[Optional["_models.OperationStatusResponse"]]
          error_map = {
              401: ClientAuthenticationError, 404: ResourceNotFoundError, 409: ResourceExistsError
          }
          error_map.update(kwargs.pop('error_map', {}))
  
          
          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,
              template_url=self._reimage_all_initial.metadata['url'],
          )
          request = _convert_request(request)
          request.url = self._client.format_url(request.url)
  
          pipeline_response = self._client._pipeline.run(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)
  
          deserialized = None
          if response.status_code == 200:
              deserialized = self._deserialize('OperationStatusResponse', pipeline_response)
  
          if cls:
              return cls(pipeline_response, deserialized, {})
  
          return deserialized
  
      _reimage_all_initial.metadata = {'url': '/subscriptions/{subscriptionId}/resourceGroups/{resourceGroupName}/providers/Microsoft.Compute/virtualMachineScaleSets/{vmScaleSetName}/reimageall'}  # type: ignore
  
  
      @distributed_trace
      def begin_reimage_all(
          self,
          resource_group_name: str,
          vm_scale_set_name: str,
          **kwargs: Any
      ) -> LROPoller["_models.OperationStatusResponse"]:
          """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
          :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 ARMPolling. 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.PollingMethod
          :keyword int polling_interval: Default waiting time between two polls for LRO operations if no
           Retry-After header is present.
          :return: An instance of LROPoller that returns either OperationStatusResponse or the result of
           cls(response)
          :rtype:
           ~azure.core.polling.LROPoller[~azure.mgmt.compute.v2016_04_30_preview.models.OperationStatusResponse]
          :raises: ~azure.core.exceptions.HttpResponseError
          """
          polling = kwargs.pop('polling', True)  # type: Union[bool, azure.core.polling.PollingMethod]
          cls = kwargs.pop('cls', None)  # type: ClsType["_models.OperationStatusResponse"]
          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 = self._reimage_all_initial(
                  resource_group_name=resource_group_name,
                  vm_scale_set_name=vm_scale_set_name,
                  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('OperationStatusResponse', pipeline_response)
              if cls:
                  return cls(pipeline_response, deserialized, {})
              return deserialized
  
  
          if polling is True: polling_method = ARMPolling(lro_delay, lro_options={'final-state-via': 'azure-async-operation'}, **kwargs)
          elif polling is False: polling_method = NoPolling()
          else: polling_method = polling
          if cont_token:
              return LROPoller.from_continuation_token(
                  polling_method=polling_method,
                  continuation_token=cont_token,
                  client=self._client,
                  deserialization_callback=get_long_running_output
              )
          else:
              return LROPoller(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
  