compute models

lib/ansible/modules/cloud/azure/azure_rm_availabilityset.py

@@ -112,9 +112,6 @@ from ansible.module_utils.azure_rm_common import AzureRMModuleBase
 
 try:
     from msrestazure.azure_exceptions import CloudError
-    from azure.mgmt.compute.models import (
-        AvailabilitySet, Sku
-    )
 except ImportError:
     # This is handled in azure_rm_common
     pass
@@ -260,10 +257,10 @@ class AzureRMAvailabilitySet(AzureRMModuleBase):
         '''
         self.log("Creating availabilityset {0}".format(self.name))
         try:
-            params_sku = Sku(
+            params_sku = self.compute_models.Sku(
                 name=self.sku
             )
-            params = AvailabilitySet(
+            params = self.compute_models.AvailabilitySet(
                 location=self.location,
                 tags=self.tags,
                 platform_update_domain_count=self.platform_update_domain_count,