Skip to content

Commit

Permalink
[Batch] Data plane SDK v12.0.0 (#22730)
Browse files Browse the repository at this point in the history
* updating library for 2022-01-015 changes

* update to tests

* new test recordings

* fixing tests

* updating vm

* updating changelong

* updating doc links

* updates based on feedback
  • Loading branch information
wiboris authored Feb 1, 2022
1 parent fba4c81 commit c506edc
Show file tree
Hide file tree
Showing 32 changed files with 5,157 additions and 2,511 deletions.
18 changes: 17 additions & 1 deletion sdk/batch/azure-batch/CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,22 @@
# Release History

## 11.0.1 (Unreleased)
## 12.0.0 (2022-02-01)

### Features

- Added property uploadHeaders to `OutputFileBlobContainerDestination`.
- Allows users to set custom HTTP headers on resource file uploads.
- Array of type HttpHeader (also being added).
- Added boolean property `allow_task_preemption` to `JobSpecification`, `CloudJob`, `JobAddParameter`, `JobPatchParameter`, `JobUpdateParameter`
- Mark Tasks as preemptible for higher priority Tasks (requires Comms-Enabled or Single Tenant Pool).
- Replaced comment (title, description, etc.) references of "low-priority" with "Spot/Low-Priority", to reflect new service behavior.
- No API change required.
- Low-Priority Compute Nodes (VMs) will continue to be used for User Subscription pools (and only User Subscription pools), as before.
- Spot Compute Nodes (VMs) will now be used for Batch Managed (and only Batch Managed pools) pools.
- Relevant docs:
https://docs.microsoft.com/azure/batch/nodes-and-pools
- https://docs.microsoft.com/azure/batch/batch-spot-vms


### Features Added

Expand Down
4 changes: 2 additions & 2 deletions sdk/batch/azure-batch/azure/batch/_batch_service_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ def __init__(
super(BatchServiceClient, 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 = '2021-06-01.14.0'
self.api_version = '2022-01-01.15.0'
self._serialize = Serializer(client_models)
self._deserialize = Deserializer(client_models)

Expand All @@ -92,4 +92,4 @@ def __init__(
self.compute_node_extension = ComputeNodeExtensionOperations(
self._client, self.config, self._serialize, self._deserialize)

patch_client()
patch_client()
2 changes: 1 addition & 1 deletion sdk/batch/azure-batch/azure/batch/_version.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,5 +9,5 @@
# regenerated.
# --------------------------------------------------------------------------

VERSION = "11.0.1"
VERSION = "12.0.0"

3 changes: 3 additions & 0 deletions sdk/batch/azure-batch/azure/batch/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,7 @@
from ._models_py3 import FileListFromComputeNodeOptions
from ._models_py3 import FileListFromTaskOptions
from ._models_py3 import FileProperties
from ._models_py3 import HttpHeader
from ._models_py3 import ImageInformation
from ._models_py3 import ImageReference
from ._models_py3 import InboundEndpoint
Expand Down Expand Up @@ -307,6 +308,7 @@
from ._models import FileListFromComputeNodeOptions
from ._models import FileListFromTaskOptions
from ._models import FileProperties
from ._models import HttpHeader
from ._models import ImageInformation
from ._models import ImageReference
from ._models import InboundEndpoint
Expand Down Expand Up @@ -595,6 +597,7 @@
'FileListFromComputeNodeOptions',
'FileListFromTaskOptions',
'FileProperties',
'HttpHeader',
'ImageInformation',
'ImageReference',
'InboundEndpoint',
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -295,7 +295,7 @@ class ComputeNodeState(str, Enum):
unknown = "unknown" #: The Batch service has lost contact with the Compute Node, and does not know its true state.
leaving_pool = "leavingpool" #: The Compute Node is leaving the Pool, either because the user explicitly removed it or because the Pool is resizing or autoscaling down.
offline = "offline" #: The Compute Node is not currently running a Task, and scheduling of new Tasks to the Compute Node is disabled.
preempted = "preempted" #: The low-priority Compute Node has been preempted. Tasks which were running on the Compute Node when it was preempted will be rescheduled when another Compute Node becomes available.
preempted = "preempted" #: The Spot/Low-priority Compute Node has been preempted. Tasks which were running on the Compute Node when it was preempted will be rescheduled when another Compute Node becomes available.


class SchedulingState(str, Enum):
Expand Down
132 changes: 111 additions & 21 deletions sdk/batch/azure-batch/azure/batch/models/_models.py

Large diffs are not rendered by default.

142 changes: 116 additions & 26 deletions sdk/batch/azure-batch/azure/batch/models/_models_py3.py

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class AccountOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class ApplicationOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class CertificateOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class ComputeNodeExtensionOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class ComputeNodeOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class FileOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class JobOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class JobScheduleOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class PoolOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ class TaskOperations(object):
:param config: Configuration of service client.
:param serializer: An object model serializer.
:param deserializer: An object model deserializer.
:ivar api_version: The API version to use for the request. Constant value: "2021-06-01.14.0".
:ivar api_version: The API version to use for the request. Constant value: "2022-01-01.15.0".
"""

models = models
Expand All @@ -34,7 +34,7 @@ def __init__(self, client, config, serializer, deserializer):
self._client = client
self._serialize = serializer
self._deserialize = deserializer
self.api_version = "2021-06-01.14.0"
self.api_version = "2022-01-01.15.0"

self.config = config

Expand Down
6 changes: 3 additions & 3 deletions sdk/batch/azure-batch/tests/batch_preparers.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,7 +183,7 @@ def create_resource(self, name, **kwargs):
group = self._get_resource_group(**kwargs)
batch_account = self._get_batch_account(**kwargs)
user = models.UserAccount(name='task-user', password='kt#_gahr!@aGERDXA', elevation_level=models.ElevationLevel.admin)
vm_size = 'Standard_D1_v2'
vm_size = 'standard_d2_v2'

if self.config == 'paas':
vm_size = 'small'
Expand All @@ -205,9 +205,9 @@ def create_resource(self, name, **kwargs):
image_reference=models.ImageReference(
publisher='Canonical',
offer='UbuntuServer',
sku='16.04-LTS'
sku='18.04-LTS'
),
node_agent_sku_id='batch.node.ubuntu 16.04'))
node_agent_sku_id='batch.node.ubuntu 18.04'))
parameters = models.Pool(
display_name="test_pool",
vm_size=vm_size,
Expand Down
Loading

0 comments on commit c506edc

Please sign in to comment.