Generated from 497f37d011e45d90e7a5f190b1326ec89e2719c8

Merge pull request #6 from Khushboo-Baheti/master

merge

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

@@ -13,15 +13,20 @@
 from msrest import Serializer, Deserializer
 
 from ._configuration import AzureMachineLearningWorkspacesConfiguration
+from .operations import AzureMachineLearningWorkspacesOperationsMixin
 from .operations import Operations
 from .operations import WorkspacesOperations
+from .operations import WorkspaceFeaturesOperations
 from .operations import UsagesOperations
 from .operations import VirtualMachineSizesOperations
+from .operations import QuotasOperations
 from .operations import MachineLearningComputeOperations
+from .operations import PrivateEndpointConnectionsOperations
+from .operations import PrivateLinkResourcesOperations
 from . import models
 
 
-class AzureMachineLearningWorkspaces(SDKClient):
+class AzureMachineLearningWorkspaces(AzureMachineLearningWorkspacesOperationsMixin, SDKClient):
     """These APIs allow end users to operate on Azure Machine Learning Workspace resources.
 
     :ivar config: Configuration for client.
@@ -31,12 +36,20 @@ class AzureMachineLearningWorkspaces(SDKClient):
     :vartype operations: azure.mgmt.machinelearningservices.operations.Operations
     :ivar workspaces: Workspaces operations
     :vartype workspaces: azure.mgmt.machinelearningservices.operations.WorkspacesOperations
+    :ivar workspace_features: WorkspaceFeatures operations
+    :vartype workspace_features: azure.mgmt.machinelearningservices.operations.WorkspaceFeaturesOperations
     :ivar usages: Usages operations
     :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
+    :ivar private_endpoint_connections: PrivateEndpointConnections operations
+    :vartype private_endpoint_connections: azure.mgmt.machinelearningservices.operations.PrivateEndpointConnectionsOperations
+    :ivar private_link_resources: PrivateLinkResources operations
+    :vartype private_link_resources: azure.mgmt.machinelearningservices.operations.PrivateLinkResourcesOperations
 
     :param credentials: Credentials needed for the client to connect to Azure.
     :type credentials: :mod:`A msrestazure Credentials
@@ -53,17 +66,25 @@ 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 = '2020-01-01'
         self._serialize = Serializer(client_models)
         self._deserialize = Deserializer(client_models)
 
         self.operations = Operations(
             self._client, self.config, self._serialize, self._deserialize)
         self.workspaces = WorkspacesOperations(
             self._client, self.config, self._serialize, self._deserialize)
+        self.workspace_features = WorkspaceFeaturesOperations(
+            self._client, self.config, self._serialize, self._deserialize)
         self.usages = UsagesOperations(
             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)
+        self.private_endpoint_connections = PrivateEndpointConnectionsOperations(
+            self._client, self.config, self._serialize, self._deserialize)
+        self.private_link_resources = PrivateLinkResourcesOperations(
+            self._client, self.config, self._serialize, self._deserialize)

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

@@ -18,6 +18,7 @@
     from ._models_py3 import AmlComputeNodeInformation
     from ._models_py3 import AmlComputeNodesInformation
     from ._models_py3 import AmlComputeProperties
+    from ._models_py3 import AmlUserFeature
     from ._models_py3 import ClusterUpdateParameters
     from ._models_py3 import Compute
     from ._models_py3 import ComputeNodesInformation
@@ -29,24 +30,42 @@
     from ._models_py3 import DataFactory
     from ._models_py3 import DataLakeAnalytics
     from ._models_py3 import DataLakeAnalyticsProperties
+    from ._models_py3 import EncryptionProperty
     from ._models_py3 import ErrorDetail
-    from ._models_py3 import ErrorResponse
+    from ._models_py3 import ErrorResponse, ErrorResponseException
     from ._models_py3 import HDInsight
     from ._models_py3 import HDInsightProperties
     from ._models_py3 import Identity
+    from ._models_py3 import KeyVaultProperties
     from ._models_py3 import ListWorkspaceKeysResult
     from ._models_py3 import MachineLearningServiceError, MachineLearningServiceErrorException
     from ._models_py3 import NodeStateCounts
     from ._models_py3 import Operation
     from ._models_py3 import OperationDisplay
     from ._models_py3 import Password
+    from ._models_py3 import PrivateEndpoint
+    from ._models_py3 import PrivateEndpointConnection
+    from ._models_py3 import PrivateLinkResource
+    from ._models_py3 import PrivateLinkResourceListResult
+    from ._models_py3 import PrivateLinkServiceConnectionState
+    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 ResourceSkuLocationInfo
+    from ._models_py3 import ResourceSkuZoneDetails
+    from ._models_py3 import Restriction
     from ._models_py3 import ScaleSettings
     from ._models_py3 import ServicePrincipalCredentials
+    from ._models_py3 import Sku
+    from ._models_py3 import SKUCapability
     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
@@ -57,6 +76,7 @@
     from ._models_py3 import VirtualMachineSizeListResult
     from ._models_py3 import VirtualMachineSshCredentials
     from ._models_py3 import Workspace
+    from ._models_py3 import WorkspaceSku
     from ._models_py3 import WorkspaceUpdateParameters
 except (SyntaxError, ImportError):
     from ._models import AKS
@@ -67,6 +87,7 @@
     from ._models import AmlComputeNodeInformation
     from ._models import AmlComputeNodesInformation
     from ._models import AmlComputeProperties
+    from ._models import AmlUserFeature
     from ._models import ClusterUpdateParameters
     from ._models import Compute
     from ._models import ComputeNodesInformation
@@ -78,24 +99,42 @@
     from ._models import DataFactory
     from ._models import DataLakeAnalytics
     from ._models import DataLakeAnalyticsProperties
+    from ._models import EncryptionProperty
     from ._models import ErrorDetail
-    from ._models import ErrorResponse
+    from ._models import ErrorResponse, ErrorResponseException
     from ._models import HDInsight
     from ._models import HDInsightProperties
     from ._models import Identity
+    from ._models import KeyVaultProperties
     from ._models import ListWorkspaceKeysResult
     from ._models import MachineLearningServiceError, MachineLearningServiceErrorException
     from ._models import NodeStateCounts
     from ._models import Operation
     from ._models import OperationDisplay
     from ._models import Password
+    from ._models import PrivateEndpoint
+    from ._models import PrivateEndpointConnection
+    from ._models import PrivateLinkResource
+    from ._models import PrivateLinkResourceListResult
+    from ._models import PrivateLinkServiceConnectionState
+    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 ResourceSkuLocationInfo
+    from ._models import ResourceSkuZoneDetails
+    from ._models import Restriction
     from ._models import ScaleSettings
     from ._models import ServicePrincipalCredentials
+    from ._models import Sku
+    from ._models import SKUCapability
     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
@@ -106,18 +145,30 @@
     from ._models import VirtualMachineSizeListResult
     from ._models import VirtualMachineSshCredentials
     from ._models import Workspace
+    from ._models import WorkspaceSku
     from ._models import WorkspaceUpdateParameters
+from ._paged_models import AmlUserFeaturePaged
 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 ._paged_models import WorkspaceSkuPaged
 from ._azure_machine_learning_workspaces_enums import (
     ProvisioningState,
+    EncryptionStatus,
     UsageUnit,
+    QuotaUnit,
+    Status,
     ResourceIdentityType,
     VmPriority,
+    RemoteLoginPortPublicAccess,
     AllocationState,
+    NodeState,
     ComputeType,
+    ReasonCode,
+    PrivateEndpointServiceConnectionStatus,
+    PrivateEndpointConnectionProvisioningState,
     UnderlyingResourceAction,
 )
 
@@ -130,6 +181,7 @@
     'AmlComputeNodeInformation',
     'AmlComputeNodesInformation',
     'AmlComputeProperties',
+    'AmlUserFeature',
     'ClusterUpdateParameters',
     'Compute',
     'ComputeNodesInformation',
@@ -141,24 +193,42 @@
     'DataFactory',
     'DataLakeAnalytics',
     'DataLakeAnalyticsProperties',
+    'EncryptionProperty',
     'ErrorDetail',
-    'ErrorResponse',
+    'ErrorResponse', 'ErrorResponseException',
     'HDInsight',
     'HDInsightProperties',
     'Identity',
+    'KeyVaultProperties',
     'ListWorkspaceKeysResult',
     'MachineLearningServiceError', 'MachineLearningServiceErrorException',
     'NodeStateCounts',
     'Operation',
     'OperationDisplay',
     'Password',
+    'PrivateEndpoint',
+    'PrivateEndpointConnection',
+    'PrivateLinkResource',
+    'PrivateLinkResourceListResult',
+    'PrivateLinkServiceConnectionState',
+    'QuotaBaseProperties',
+    'QuotaUpdateParameters',
     'RegistryListCredentialsResult',
     'Resource',
     'ResourceId',
+    'ResourceName',
+    'ResourceQuota',
+    'ResourceSkuLocationInfo',
+    'ResourceSkuZoneDetails',
+    'Restriction',
     'ScaleSettings',
     'ServicePrincipalCredentials',
+    'Sku',
+    'SKUCapability',
     'SslConfiguration',
     'SystemService',
+    'UpdateWorkspaceQuotas',
+    'UpdateWorkspaceQuotasResult',
     'Usage',
     'UsageName',
     'UserAccountCredentials',
@@ -169,16 +239,28 @@
     'VirtualMachineSizeListResult',
     'VirtualMachineSshCredentials',
     'Workspace',
+    'WorkspaceSku',
     'WorkspaceUpdateParameters',
     'OperationPaged',
     'WorkspacePaged',
+    'AmlUserFeaturePaged',
     'UsagePaged',
+    'ResourceQuotaPaged',
     'ComputeResourcePaged',
+    'WorkspaceSkuPaged',
     'ProvisioningState',
+    'EncryptionStatus',
     'UsageUnit',
+    'QuotaUnit',
+    'Status',
     'ResourceIdentityType',
     'VmPriority',
+    'RemoteLoginPortPublicAccess',
     'AllocationState',
+    'NodeState',
     'ComputeType',
+    'ReasonCode',
+    'PrivateEndpointServiceConnectionStatus',
+    'PrivateEndpointConnectionProvisioningState',
     'UnderlyingResourceAction',
 ]

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

@@ -23,11 +23,34 @@ class ProvisioningState(str, Enum):
     canceled = "Canceled"
 
 
+class EncryptionStatus(str, Enum):
+
+    enabled = "Enabled"
+    disabled = "Disabled"
+
+
 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"
+    operation_not_supported_for_sku = "OperationNotSupportedForSku"
+    operation_not_enabled_for_region = "OperationNotEnabledForRegion"
+
+
 class ResourceIdentityType(str, Enum):
 
     system_assigned = "SystemAssigned"
@@ -39,12 +62,29 @@ class VmPriority(str, Enum):
     low_priority = "LowPriority"
 
 
+class RemoteLoginPortPublicAccess(str, Enum):
+
+    enabled = "Enabled"
+    disabled = "Disabled"
+    not_specified = "NotSpecified"
+
+
 class AllocationState(str, Enum):
 
     steady = "Steady"
     resizing = "Resizing"
 
 
+class NodeState(str, Enum):
+
+    idle = "idle"
+    running = "running"
+    preparing = "preparing"
+    unusable = "unusable"
+    leaving = "leaving"
+    preempted = "preempted"
+
+
 class ComputeType(str, Enum):
 
     aks = "AKS"
@@ -56,6 +96,28 @@ class ComputeType(str, Enum):
     data_lake_analytics = "DataLakeAnalytics"
 
 
+class ReasonCode(str, Enum):
+
+    not_specified = "NotSpecified"
+    not_available_for_region = "NotAvailableForRegion"
+    not_available_for_subscription = "NotAvailableForSubscription"
+
+
+class PrivateEndpointServiceConnectionStatus(str, Enum):
+
+    pending = "Pending"
+    approved = "Approved"
+    rejected = "Rejected"
+
+
+class PrivateEndpointConnectionProvisioningState(str, Enum):
+
+    succeeded = "Succeeded"
+    creating = "Creating"
+    deleting = "Deleting"
+    failed = "Failed"
+
+
 class UnderlyingResourceAction(str, Enum):
 
     delete = "Delete"