Skip to content

Commit 118e99f

Browse files
authored
[ML] Fix add missing "properties" to batch deployment (Azure#27292)
1 parent a10fb9c commit 118e99f

File tree

5 files changed

+1266
-1306
lines changed

5 files changed

+1266
-1306
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/entities/_deployment/batch_deployment.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ def _to_rest_object(self, location: str) -> BatchDeploymentData: # pylint: disa
208208
mini_batch_size=self.mini_batch_size,
209209
max_concurrency_per_instance=self.max_concurrency_per_instance,
210210
environment_variables=self.environment_variables,
211+
properties=self.properties,
211212
)
212213

213214
return BatchDeploymentData(location=location, properties=batch_deployment, tags=self.tags)
@@ -243,6 +244,7 @@ def _from_rest_object(cls, deployment: BatchDeploymentData): # pylint: disable=
243244
environment_variables=deployment.properties.environment_variables,
244245
max_concurrency_per_instance=deployment.properties.max_concurrency_per_instance,
245246
endpoint_name=_parse_endpoint_name_from_deployment_id(deployment.id),
247+
properties=deployment.properties.properties,
246248
)
247249

248250
@classmethod

sdk/ml/azure-ai-ml/tests/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def fake_datastore_key() -> str:
5959

6060
@pytest.fixture(autouse=True)
6161
def add_sanitizers(test_proxy, fake_datastore_key):
62-
add_remove_header_sanitizer(headers="x-azureml-token,Log-URL")
62+
add_remove_header_sanitizer(headers="x-azureml-token,Log-URL,Authorization")
6363
set_custom_default_matcher(
6464
excluded_headers="x-ms-meta-name,x-ms-meta-version", ignored_query_parameters="api-version"
6565
)

0 commit comments

Comments
 (0)