fix data_id showing (Azure#30202)

Author: banibrata-de

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/online/deployment_collection_schema.py

@@ -5,9 +5,9 @@
 import logging
 from typing import Any
 
-from marshmallow import post_load
+from marshmallow import post_load, fields
 
-from azure.ai.ml._schema import PatchedSchemaMeta, StringTransformedEnum, NestedField
+from azure.ai.ml._schema import PatchedSchemaMeta, StringTransformedEnum, NestedField, UnionField
 from azure.ai.ml._schema._deployment.online.data_asset_schema import DataAssetSchema
 from azure.ai.ml.constants._common import Boolean
 
@@ -16,7 +16,13 @@
 
 class DeploymentCollectionSchema(metaclass=PatchedSchemaMeta):
     enabled = StringTransformedEnum(required=True, allowed_values=[Boolean.TRUE, Boolean.FALSE])
-    data = NestedField(DataAssetSchema)
+    data = UnionField(
+        [
+            fields.Str(),
+            NestedField(DataAssetSchema),
+        ]
+    )
+    client_id = fields.Str()
 
     # pylint: disable=unused-argument,no-self-use
     @post_load

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/_deployment/online/online_deployment.py

@@ -68,6 +68,7 @@ class ManagedOnlineDeploymentSchema(OnlineDeploymentSchema):
         allowed_values=[PublicNetworkAccess.ENABLED, PublicNetworkAccess.DISABLED]
     )
     private_network_connection = ExperimentalField(fields.Bool())
+    data_collector = NestedField(DataCollectorSchema)
 
     @post_load
     def make(self, data: Any, **kwargs: Any) -> Any:

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

@@ -630,6 +630,11 @@ class ManagedOnlineDeployment(OnlineDeployment):
     :keyword code_path: Equivalent to code_configuration.code, will be ignored if code_configuration is present
         , defaults to None
     :paramtype code_path: typing.Optional[typing.Union[str, os.PathLike]]
+    :keyword scoring_script_path: Equivalent to code_configuration.scoring_script, will be ignored if
+        code_configuration is present, defaults to None
+    :paramtype scoring_script_path: typing.Optional[typing.Union[str, os.PathLike]]
+    :keyword data_collector: Data collector, defaults to None
+    :paramtype data_collectors: typing.Optional[typing.List[~azure.ai.ml.entities.DataCollector]]
     """
 
     def __init__(
@@ -656,6 +661,7 @@ def __init__(
         scoring_script: Optional[
             Union[str, os.PathLike]
         ] = None,  # promoted property from code_configuration.scoring_script
+        data_collector: Optional[DataCollector] = None,
         **kwargs,
     ):
         """Managed Online endpoint deployment entity.
@@ -724,6 +730,7 @@ def __init__(
             instance_type=instance_type,
             code_path=code_path,
             scoring_script=scoring_script,
+            data_collector=data_collector,
             **kwargs,
         )