[AutoPR azure-mgmt-batch] Enable Batch python track2 pipeline

sdk/batch/azure-mgmt-batch/azure/mgmt/batch/_batch_management_client.py

@@ -66,7 +66,7 @@ def __init__(
         super(BatchManagementClient, 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 = '2020-05-01'
+        self.api_version = '2021-01-01'
         self._serialize = Serializer(client_models)
         self._deserialize = Deserializer(client_models)
 

sdk/batch/azure-mgmt-batch/azure/mgmt/batch/models/__init__.py

@@ -25,10 +25,13 @@
     from ._models_py3 import BatchAccount
     from ._models_py3 import BatchAccountCreateParameters
     from ._models_py3 import BatchAccountIdentity
+    from ._models_py3 import BatchAccountIdentityUserAssignedIdentitiesValue
     from ._models_py3 import BatchAccountKeys
     from ._models_py3 import BatchAccountRegenerateKeyParameters
     from ._models_py3 import BatchAccountUpdateParameters
     from ._models_py3 import BatchLocationQuota
+    from ._models_py3 import BatchPoolIdentity
+    from ._models_py3 import BatchPoolIdentityUserAssignedIdentitiesValue
     from ._models_py3 import Certificate
     from ._models_py3 import CertificateBaseProperties
     from ._models_py3 import CertificateCreateOrUpdateParameters
@@ -56,6 +59,7 @@
     from ._models_py3 import NetworkConfiguration
     from ._models_py3 import NetworkSecurityGroupRule
     from ._models_py3 import NFSMountConfiguration
+    from ._models_py3 import NodePlacementConfiguration
     from ._models_py3 import Operation
     from ._models_py3 import OperationDisplay
     from ._models_py3 import Pool
@@ -78,6 +82,7 @@
     from ._models_py3 import UserIdentity
     from ._models_py3 import VirtualMachineConfiguration
     from ._models_py3 import VirtualMachineFamilyCoreQuota
+    from ._models_py3 import VMExtension
     from ._models_py3 import WindowsConfiguration
     from ._models_py3 import WindowsUserConfiguration
 except (SyntaxError, ImportError):
@@ -96,10 +101,13 @@
     from ._models import BatchAccount
     from ._models import BatchAccountCreateParameters
     from ._models import BatchAccountIdentity
+    from ._models import BatchAccountIdentityUserAssignedIdentitiesValue
     from ._models import BatchAccountKeys
     from ._models import BatchAccountRegenerateKeyParameters
     from ._models import BatchAccountUpdateParameters
     from ._models import BatchLocationQuota
+    from ._models import BatchPoolIdentity
+    from ._models import BatchPoolIdentityUserAssignedIdentitiesValue
     from ._models import Certificate
     from ._models import CertificateBaseProperties
     from ._models import CertificateCreateOrUpdateParameters
@@ -127,6 +135,7 @@
     from ._models import NetworkConfiguration
     from ._models import NetworkSecurityGroupRule
     from ._models import NFSMountConfiguration
+    from ._models import NodePlacementConfiguration
     from ._models import Operation
     from ._models import OperationDisplay
     from ._models import Pool
@@ -149,6 +158,7 @@
     from ._models import UserIdentity
     from ._models import VirtualMachineConfiguration
     from ._models import VirtualMachineFamilyCoreQuota
+    from ._models import VMExtension
     from ._models import WindowsConfiguration
     from ._models import WindowsUserConfiguration
 from ._paged_models import ApplicationPackagePaged
@@ -176,6 +186,7 @@
     CachingType,
     StorageAccountType,
     DiskEncryptionTarget,
+    NodePlacementPolicyType,
     ComputeNodeDeallocationOption,
     InterNodeCommunicationState,
     InboundEndpointProtocol,
@@ -188,8 +199,7 @@
     ContainerWorkingDirectory,
     CertificateStoreLocation,
     CertificateVisibility,
-    ContainerType,
-    ResourceType,
+    PoolIdentityType,
     NameAvailabilityReason,
 )
 
@@ -209,10 +219,13 @@
     'BatchAccount',
     'BatchAccountCreateParameters',
     'BatchAccountIdentity',
+    'BatchAccountIdentityUserAssignedIdentitiesValue',
     'BatchAccountKeys',
     'BatchAccountRegenerateKeyParameters',
     'BatchAccountUpdateParameters',
     'BatchLocationQuota',
+    'BatchPoolIdentity',
+    'BatchPoolIdentityUserAssignedIdentitiesValue',
     'Certificate',
     'CertificateBaseProperties',
     'CertificateCreateOrUpdateParameters',
@@ -240,6 +253,7 @@
     'NetworkConfiguration',
     'NetworkSecurityGroupRule',
     'NFSMountConfiguration',
+    'NodePlacementConfiguration',
     'Operation',
     'OperationDisplay',
     'Pool',
@@ -262,6 +276,7 @@
     'UserIdentity',
     'VirtualMachineConfiguration',
     'VirtualMachineFamilyCoreQuota',
+    'VMExtension',
     'WindowsConfiguration',
     'WindowsUserConfiguration',
     'BatchAccountPaged',
@@ -288,6 +303,7 @@
     'CachingType',
     'StorageAccountType',
     'DiskEncryptionTarget',
+    'NodePlacementPolicyType',
     'ComputeNodeDeallocationOption',
     'InterNodeCommunicationState',
     'InboundEndpointProtocol',
@@ -300,7 +316,6 @@
     'ContainerWorkingDirectory',
     'CertificateStoreLocation',
     'CertificateVisibility',
-    'ContainerType',
-    'ResourceType',
+    'PoolIdentityType',
     'NameAvailabilityReason',
 ]

sdk/batch/azure-mgmt-batch/azure/mgmt/batch/models/_batch_management_client_enums.py

@@ -33,6 +33,7 @@ class PublicNetworkAccessType(str, Enum):
 class ResourceIdentityType(str, Enum):
 
     system_assigned = "SystemAssigned"  #: Batch account has a system assigned identity with it.
+    user_assigned = "UserAssigned"  #: Batch account has user assigned identities with it.
     none = "None"  #: Batch account has no identity associated with it. Setting `None` in update account will remove existing identities.
 
 
@@ -118,6 +119,12 @@ class DiskEncryptionTarget(str, Enum):
     temporary_disk = "TemporaryDisk"  #: The temporary disk on the compute node is encrypted. On Linux this encryption applies to other partitions (such as those on mounted data disks) when encryption occurs at boot time.
 
 
+class NodePlacementPolicyType(str, Enum):
+
+    regional = "Regional"  #: All nodes in the pool will be allocated in the same region.
+    zonal = "Zonal"  #: Nodes in the pool will be spread across different zones with best effort balancing.
+
+
 class ComputeNodeDeallocationOption(str, Enum):
 
     requeue = "Requeue"  #: Terminate running task processes and requeue the tasks. The tasks will run again when a node is available. Remove nodes as soon as tasks have been terminated.
@@ -154,7 +161,7 @@ class IPAddressProvisioningType(str, Enum):
 class ComputeNodeFillType(str, Enum):
 
     spread = "Spread"  #: Tasks should be assigned evenly across all nodes in the pool.
-    pack = "Pack"  #: As many tasks as possible (maxTasksPerNode) should be assigned to each node in the pool before any tasks are assigned to the next node in the pool.
+    pack = "Pack"  #: As many tasks as possible (taskSlotsPerNode) should be assigned to each node in the pool before any tasks are assigned to the next node in the pool.
 
 
 class ElevationLevel(str, Enum):
@@ -194,14 +201,10 @@ class CertificateVisibility(str, Enum):
     remote_user = "RemoteUser"  #: The certificate should be visible to the user accounts under which users remotely access the node.
 
 
-class ContainerType(str, Enum):
-
-    docker_compatible = "DockerCompatible"  #: A Docker compatible container technology will be used to launch the containers.
-
-
-class ResourceType(str, Enum):
+class PoolIdentityType(str, Enum):
 
-    batch_accounts = "Microsoft.Batch/batchAccounts"  #: The Batch account resource type.
+    user_assigned = "UserAssigned"  #: Batch pool has user assigned identities with it.
+    none = "None"  #: Batch pool has no identity associated with it. Setting `None` in update pool will remove existing identities.
 
 
 class NameAvailabilityReason(str, Enum):