From 247d41fe8e3fa0b64a25994ed3da3572d63bab12 Mon Sep 17 00:00:00 2001 From: nemanjarajic Date: Tue, 6 Feb 2024 14:42:31 -0800 Subject: [PATCH] Batch endpoint fix serialization error --- sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py | 2 +- .../ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py index 345d4661cb29..0e3b3c7242ad 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_ml_client.py @@ -521,7 +521,7 @@ def __init__( self._batch_endpoints = BatchEndpointOperations( self._operation_scope, self._operation_config, - self._service_client_05_2022, + self._service_client_10_2023, self._operation_container, self._credential, requests_pipeline=self._requests_pipeline, diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py index db4c986eb0a1..49699bb04416 100644 --- a/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py +++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_endpoint/batch/batch_endpoint_defaults.py @@ -9,7 +9,7 @@ from marshmallow import fields, post_load -from azure.ai.ml._restclient.v2022_05_01.models import BatchEndpointDefaults +from azure.ai.ml._restclient.v2023_10_01.models import BatchEndpointDefaults from azure.ai.ml._schema.core.schema import PatchedSchemaMeta module_logger = logging.getLogger(__name__)