[Batch] Beta Data Plane SDK 15.0.0b2 Update to 07-01-2024 API (Azure#40222)

* Generate off latest TypeSpec for 2024-07-01 API changes

* Update timeout naming

* Regenerate for long encoding update

* Fix unit tests

* More changes from long encoding update

* Fix unit tests

* Fix typing errors

* Async test helpers now preserve type info

* Add test utility for handling file downloads

* Update async wrapper

* Re-generate with certificates re-added in

* Update recording assets

* Fix mypy errors

* Minor space add

* Mypy fixes and update tests

* Update changelog

* Fix test_batch_files by adding BatchFileProperties
- made manual operation naming changes that need to be added to typespec

* Update tests

* Fixes

---------

Co-authored-by: Rena Chen <[email protected]>
Co-authored-by: David Watrous <[email protected]>

Author: 3 people

sdk/batch/azure-batch/CHANGELOG.md

@@ -1,14 +1,37 @@
 # Release History
 
-## 15.0.0b2 (Unreleased)
+## 15.0.0b2 (2025-03-01)
 
 ### Features Added
 
-### Breaking Changes
+- Force delete/terminate job or job schedule:
+  - Added `force` parameter of type Boolean to `delete_job`, `terminate_job`,  `delete_job_schedule`, and `terminate_job_schedule`
 
-### Bugs Fixed
+- Support for compute node start/deallocate operations:
+  - Added `start_node`, `deallocate_node` methods to `BatchClient` and `AsyncBatchClient`
 
-### Other Changes
+- Container task data mount isolation:
+  - Added `containerHostBatchBindMounts` of type `List<ContainerHostBatchBindMountEntry>` to `BatchTaskContainerSettings`
+
+- Patch improvements for pool and job:
+  - Added `displayName`, `vmSize`, `taskSlotsPerNode`, `taskSchedulingPolicy`, `enableInterNodeCommunication`, `virtualMachineConfiguration`, `networkConfiguration`, `userAccounts`, `mountConfiguration`, `upgradePolicy`, and `resourceTags` to `BatchPoolUpdateContent`
+  - Added `networkConfiguration` to `BatchJobUpdateContent`
+
+- Confidential VM support:
+  - Added `confidentialVM` to `SecurityTypes`.
+  - Added `securityProfile` of type `VMDiskSecurityProfile` to `ManagedDisk`
+
+- Support for shared and community gallery images:
+  - Added `sharedGalleryImageId` and `communityGalleryImageId` to `ImageReference`
+
+- Re-add support for `BatchCertificate` (temporary since this feature is deprecated):
+  - Added `create_certificate`, `list_certificates`, `cancel_certificate_deletion`, `delete_certificate`, and`get_certificate` methods to `BatchClient` and `AsyncBatchClient`
+
+### Breaking Changes
+
+- Removed `get_remote_desktop` method from `BatchClient`. Use `get_node_remote_login_settings` instead to remotely login to a compute node
+- Removed `CloudServiceConfiguration` from pool models and operations. Use `VirtualMachineConfiguration` when creating pools
+- Removed `ApplicationLicenses` from pool models and operations
 
 ## 15.0.0b1 (2024-09-01)
 

sdk/batch/azure-batch/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/batch/azure-batch",
-  "Tag": "python/batch/azure-batch_b2cb0e7ad7"
+  "Tag": "python/batch/azure-batch_8d51b2d53f"
 }

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

@@ -5,22 +5,28 @@
 # Code generated by Microsoft (R) Python Code Generator.
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
+# pylint: disable=wrong-import-position
 
-from ._client import BatchClient
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from ._patch import *  # pylint: disable=unused-wildcard-import
+
+from ._client import BatchClient  # type: ignore
 from ._version import VERSION
 
 __version__ = VERSION
 
 try:
     from ._patch import __all__ as _patch_all
-    from ._patch import *  # pylint: disable=unused-wildcard-import
+    from ._patch import *
 except ImportError:
     _patch_all = []
 from ._patch import patch_sdk as _patch_sdk
 
 __all__ = [
     "BatchClient",
 ]
-__all__.extend([p for p in _patch_all if p not in __all__])
+__all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 
 _patch_sdk()

sdk/batch/azure-batch/azure/batch/_client.py

@@ -8,6 +8,7 @@
 
 from copy import deepcopy
 from typing import Any, TYPE_CHECKING
+from typing_extensions import Self
 
 from azure.core import PipelineClient
 from azure.core.pipeline import policies
@@ -18,20 +19,20 @@
 from ._serialization import Deserializer, Serializer
 
 if TYPE_CHECKING:
-    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
-class BatchClient(BatchClientOperationsMixin):  # pylint: disable=client-accepts-api-version-keyword
+class BatchClient(BatchClientOperationsMixin):
     """BatchClient.
 
     :param endpoint: Batch account endpoint (for example:
-     https://batchaccount.eastus2.batch.azure.com). Required.
+     `https://batchaccount.eastus2.batch.azure.com
+     <https://batchaccount.eastus2.batch.azure.com>`_). Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Required.
     :type credential: ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
-     "2024-02-01.19.0". Note that overriding this default value may result in unsupported behavior.
+     "2024-07-01.20.0". Note that overriding this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
@@ -91,7 +92,7 @@ def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs:
     def close(self) -> None:
         self._client.close()
 
-    def __enter__(self) -> "BatchClient":
+    def __enter__(self) -> Self:
         self._client.__enter__()
         return self
 

sdk/batch/azure-batch/azure/batch/_configuration.py

@@ -13,7 +13,6 @@
 from ._version import VERSION
 
 if TYPE_CHECKING:
-    # pylint: disable=unused-import,ungrouped-imports
     from azure.core.credentials import TokenCredential
 
 
@@ -24,17 +23,18 @@ class BatchClientConfiguration:  # pylint: disable=too-many-instance-attributes
     attributes.
 
     :param endpoint: Batch account endpoint (for example:
-     https://batchaccount.eastus2.batch.azure.com). Required.
+     `https://batchaccount.eastus2.batch.azure.com
+     <https://batchaccount.eastus2.batch.azure.com>`_). Required.
     :type endpoint: str
     :param credential: Credential used to authenticate requests to the service. Required.
     :type credential: ~azure.core.credentials.TokenCredential
     :keyword api_version: The API version to use for this operation. Default value is
-     "2024-02-01.19.0". Note that overriding this default value may result in unsupported behavior.
+     "2024-07-01.20.0". Note that overriding this default value may result in unsupported behavior.
     :paramtype api_version: str
     """
 
     def __init__(self, endpoint: str, credential: "TokenCredential", **kwargs: Any) -> None:
-        api_version: str = kwargs.pop("api_version", "2024-02-01.19.0")
+        api_version: str = kwargs.pop("api_version", "2024-07-01.20.0")
 
         if endpoint is None:
             raise ValueError("Parameter 'endpoint' must not be None.")