<commit_before>from .._locale import L
from azure.mgmt.compute import ComputeManagementClient, ComputeManagementClientConfiguration
from azure.mgmt.compute.operations import (AvailabilitySetsOperations,
                                           VirtualMachineExtensionImagesOperations,
                                           VirtualMachineExtensionsOperations,
                                           VirtualMachineImagesOperations,
                                           UsageOperations,
                                           VirtualMachineSizesOperations,
                                           VirtualMachinesOperations,
                                           VirtualMachineScaleSetsOperations,
                                           VirtualMachineScaleSetVMsOperations)

from ._command_creation import get_mgmt_service_client
from ..commands._auto_command import build_operation, LongRunningOperation

def _compute_client_factory():
    return get_mgmt_service_client(ComputeManagementClient, ComputeManagementClientConfiguration)

# TODO: Remove when [#115382979] has been resolved
from azure.mgmt.compute.models import VirtualMachineExtension # pylint: disable=wrong-import-position
VirtualMachineExtension._validation['location']['required'] = False # pylint: disable=protected-access

# pylint: disable=line-too-long
build_operation("vm",
                "availabilityset",
                "availability_sets",
                _compute_client_factory,
                [
                    (AvailabilitySetsOperations.delete, None),
                    (AvailabilitySetsOperations.get, 'AvailabilitySet'),
                    (AvailabilitySetsOperations.list, '[AvailabilitySet]'),
                    (AvailabilitySetsOperations.list_available_sizes, '[VirtualMachineSize]')
                ])
<commit_msg>Remove VM Extension override that was added as a temp solution<commit_after>from .._locale import L
from azure.mgmt.compute import ComputeManagementClient, ComputeManagementClientConfiguration
from azure.mgmt.compute.operations import (AvailabilitySetsOperations,
                                           VirtualMachineExtensionImagesOperations,
                                           VirtualMachineExtensionsOperations,
                                           VirtualMachineImagesOperations,
                                           UsageOperations,
                                           VirtualMachineSizesOperations,
                                           VirtualMachinesOperations,
                                           VirtualMachineScaleSetsOperations,
                                           VirtualMachineScaleSetVMsOperations)

from ._command_creation import get_mgmt_service_client
from ..commands._auto_command import build_operation, LongRunningOperation

def _compute_client_factory():
    return get_mgmt_service_client(ComputeManagementClient, ComputeManagementClientConfiguration)

# pylint: disable=line-too-long
build_operation("vm",
                "availabilityset",
                "availability_sets",
                _compute_client_factory,
                [
                    (AvailabilitySetsOperations.delete, None),
                    (AvailabilitySetsOperations.get, 'AvailabilitySet'),
                    (AvailabilitySetsOperations.list, '[AvailabilitySet]'),
                    (AvailabilitySetsOperations.list_available_sizes, '[VirtualMachineSize]')
                ])

{'fraction_non_alphanumeric': 0.05373563218390805, 'fraction_numerical': 0.002586206896551724, 'mean_word_length': 1.9008333333333334, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 32, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6410998', 'n_tokens_mistral': 674, 'n_tokens_neox': 646, 'n_words': 125}