[AutoPR azure-mgmt-machinelearningservices] [READY FOR DESIGN REVIEW] -> MachineLearningServices: Adding new API version and definition for AmlInstance compute type

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/_azure_machine_learning_workspaces.py

@@ -17,6 +17,7 @@
 from .operations import WorkspacesOperations
 from .operations import UsagesOperations
 from .operations import VirtualMachineSizesOperations
+from .operations import QuotasOperations
 from .operations import MachineLearningComputeOperations
 from . import models
 
@@ -35,6 +36,8 @@ class AzureMachineLearningWorkspaces(SDKClient):
     :vartype usages: azure.mgmt.machinelearningservices.operations.UsagesOperations
     :ivar virtual_machine_sizes: VirtualMachineSizes operations
     :vartype virtual_machine_sizes: azure.mgmt.machinelearningservices.operations.VirtualMachineSizesOperations
+    :ivar quotas: Quotas operations
+    :vartype quotas: azure.mgmt.machinelearningservices.operations.QuotasOperations
     :ivar machine_learning_compute: MachineLearningCompute operations
     :vartype machine_learning_compute: azure.mgmt.machinelearningservices.operations.MachineLearningComputeOperations
 
@@ -53,7 +56,7 @@ def __init__(
         super(AzureMachineLearningWorkspaces, self).__init__(self.config.credentials, self.config)
 
         client_models = {k: v for k, v in models.__dict__.items() if isinstance(v, type)}
-        self.api_version = '2019-05-01'
+        self.api_version = '2019-11-01'
         self._serialize = Serializer(client_models)
         self._deserialize = Deserializer(client_models)
 
@@ -65,5 +68,7 @@ def __init__(
             self._client, self.config, self._serialize, self._deserialize)
         self.virtual_machine_sizes = VirtualMachineSizesOperations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.quotas = QuotasOperations(
+            self._client, self.config, self._serialize, self._deserialize)
         self.machine_learning_compute = MachineLearningComputeOperations(
             self._client, self.config, self._serialize, self._deserialize)

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/__init__.py

@@ -20,6 +20,12 @@
     from ._models_py3 import AmlComputeProperties
     from ._models_py3 import ClusterUpdateParameters
     from ._models_py3 import Compute
+    from ._models_py3 import ComputeInstance
+    from ._models_py3 import ComputeInstanceApplication
+    from ._models_py3 import ComputeInstanceConnectivityEndpoints
+    from ._models_py3 import ComputeInstanceCreatedBy
+    from ._models_py3 import ComputeInstanceProperties
+    from ._models_py3 import ComputeInstanceSshSettings
     from ._models_py3 import ComputeNodesInformation
     from ._models_py3 import ComputeResource
     from ._models_py3 import ComputeSecrets
@@ -40,13 +46,19 @@
     from ._models_py3 import Operation
     from ._models_py3 import OperationDisplay
     from ._models_py3 import Password
+    from ._models_py3 import QuotaBaseProperties
+    from ._models_py3 import QuotaUpdateParameters
     from ._models_py3 import RegistryListCredentialsResult
     from ._models_py3 import Resource
     from ._models_py3 import ResourceId
+    from ._models_py3 import ResourceName
+    from ._models_py3 import ResourceQuota
     from ._models_py3 import ScaleSettings
     from ._models_py3 import ServicePrincipalCredentials
     from ._models_py3 import SslConfiguration
     from ._models_py3 import SystemService
+    from ._models_py3 import UpdateWorkspaceQuotas
+    from ._models_py3 import UpdateWorkspaceQuotasResult
     from ._models_py3 import Usage
     from ._models_py3 import UsageName
     from ._models_py3 import UserAccountCredentials
@@ -69,6 +81,12 @@
     from ._models import AmlComputeProperties
     from ._models import ClusterUpdateParameters
     from ._models import Compute
+    from ._models import ComputeInstance
+    from ._models import ComputeInstanceApplication
+    from ._models import ComputeInstanceConnectivityEndpoints
+    from ._models import ComputeInstanceCreatedBy
+    from ._models import ComputeInstanceProperties
+    from ._models import ComputeInstanceSshSettings
     from ._models import ComputeNodesInformation
     from ._models import ComputeResource
     from ._models import ComputeSecrets
@@ -89,13 +107,19 @@
     from ._models import Operation
     from ._models import OperationDisplay
     from ._models import Password
+    from ._models import QuotaBaseProperties
+    from ._models import QuotaUpdateParameters
     from ._models import RegistryListCredentialsResult
     from ._models import Resource
     from ._models import ResourceId
+    from ._models import ResourceName
+    from ._models import ResourceQuota
     from ._models import ScaleSettings
     from ._models import ServicePrincipalCredentials
     from ._models import SslConfiguration
     from ._models import SystemService
+    from ._models import UpdateWorkspaceQuotas
+    from ._models import UpdateWorkspaceQuotasResult
     from ._models import Usage
     from ._models import UsageName
     from ._models import UserAccountCredentials
@@ -109,14 +133,23 @@
     from ._models import WorkspaceUpdateParameters
 from ._paged_models import ComputeResourcePaged
 from ._paged_models import OperationPaged
+from ._paged_models import ResourceQuotaPaged
 from ._paged_models import UsagePaged
 from ._paged_models import WorkspacePaged
 from ._azure_machine_learning_workspaces_enums import (
     ProvisioningState,
     UsageUnit,
+    QuotaUnit,
+    Status,
     ResourceIdentityType,
     VmPriority,
+    OsType,
+    RemoteLoginPortPublicAccess,
     AllocationState,
+    ApplicationSharingPolicy,
+    SshPublicAccess,
+    ComputeInstanceState,
+    NodeState,
     ComputeType,
     UnderlyingResourceAction,
 )
@@ -132,6 +165,12 @@
     'AmlComputeProperties',
     'ClusterUpdateParameters',
     'Compute',
+    'ComputeInstance',
+    'ComputeInstanceApplication',
+    'ComputeInstanceConnectivityEndpoints',
+    'ComputeInstanceCreatedBy',
+    'ComputeInstanceProperties',
+    'ComputeInstanceSshSettings',
     'ComputeNodesInformation',
     'ComputeResource',
     'ComputeSecrets',
@@ -152,13 +191,19 @@
     'Operation',
     'OperationDisplay',
     'Password',
+    'QuotaBaseProperties',
+    'QuotaUpdateParameters',
     'RegistryListCredentialsResult',
     'Resource',
     'ResourceId',
+    'ResourceName',
+    'ResourceQuota',
     'ScaleSettings',
     'ServicePrincipalCredentials',
     'SslConfiguration',
     'SystemService',
+    'UpdateWorkspaceQuotas',
+    'UpdateWorkspaceQuotasResult',
     'Usage',
     'UsageName',
     'UserAccountCredentials',
@@ -173,12 +218,21 @@
     'OperationPaged',
     'WorkspacePaged',
     'UsagePaged',
+    'ResourceQuotaPaged',
     'ComputeResourcePaged',
     'ProvisioningState',
     'UsageUnit',
+    'QuotaUnit',
+    'Status',
     'ResourceIdentityType',
     'VmPriority',
+    'OsType',
+    'RemoteLoginPortPublicAccess',
     'AllocationState',
+    'ApplicationSharingPolicy',
+    'SshPublicAccess',
+    'ComputeInstanceState',
+    'NodeState',
     'ComputeType',
     'UnderlyingResourceAction',
 ]

sdk/machinelearning/azure-mgmt-machinelearningservices/azure/mgmt/machinelearningservices/models/_azure_machine_learning_workspaces_enums.py

@@ -28,6 +28,21 @@ class UsageUnit(str, Enum):
     count = "Count"
 
 
+class QuotaUnit(str, Enum):
+
+    count = "Count"
+
+
+class Status(str, Enum):
+
+    undefined = "Undefined"
+    success = "Success"
+    failure = "Failure"
+    invalid_quota_below_cluster_minimum = "InvalidQuotaBelowClusterMinimum"
+    invalid_quota_exceeds_subscription_limit = "InvalidQuotaExceedsSubscriptionLimit"
+    invalid_vm_family_name = "InvalidVMFamilyName"
+
+
 class ResourceIdentityType(str, Enum):
 
     system_assigned = "SystemAssigned"
@@ -39,16 +54,72 @@ class VmPriority(str, Enum):
     low_priority = "LowPriority"
 
 
+class OsType(str, Enum):
+
+    linux = "Linux"
+    windows = "Windows"
+
+
+class RemoteLoginPortPublicAccess(str, Enum):
+
+    enabled = "Enabled"
+    disabled = "Disabled"
+    not_specified = "NotSpecified"
+
+
 class AllocationState(str, Enum):
 
     steady = "Steady"
     resizing = "Resizing"
 
 
+class ApplicationSharingPolicy(str, Enum):
+
+    personal = "Personal"
+    shared = "Shared"
+
+
+class SshPublicAccess(str, Enum):
+
+    enabled = "Enabled"
+    disabled = "Disabled"
+
+
+class ComputeInstanceState(str, Enum):
+
+    creating = "Creating"
+    create_failed = "CreateFailed"
+    deleting = "Deleting"
+    ready = "Ready"
+    restarting = "Restarting"
+    restart_failed = "RestartFailed"
+    running = "Running"
+    setting_up = "SettingUp"
+    starting = "Starting"
+    start_failed = "StartFailed"
+    stop_failed = "StopFailed"
+    stopped = "Stopped"
+    stopping = "Stopping"
+    user_setting_up = "UserSettingUp"
+    unknown = "Unknown"
+    unusable = "Unusable"
+
+
+class NodeState(str, Enum):
+
+    idle = "idle"
+    running = "running"
+    preparing = "preparing"
+    unusable = "unusable"
+    leaving = "leaving"
+    preempted = "preempted"
+
+
 class ComputeType(str, Enum):
 
     aks = "AKS"
     aml_compute = "AmlCompute"
+    compute_instance = "ComputeInstance"
     data_factory = "DataFactory"
     virtual_machine = "VirtualMachine"
     hd_insight = "HDInsight"