Skip to content

Commit e170ad4

Browse files
authored
Add creation context to online deployment from rest (#35565)
1 parent 8c0e3cd commit e170ad4

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from azure.ai.ml._schema.core.fields import ExperimentalField, NestedField, StringTransformedEnum, UnionField
1616
from azure.ai.ml._utils.utils import camel_to_snake
1717
from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY, PublicNetworkAccess
18+
from azure.ai.ml._schema.job.creation_context import CreationContextSchema
1819

1920
from .data_collector_schema import DataCollectorSchema
2021
from .liveness_probe import LivenessProbeSchema
@@ -69,6 +70,7 @@ class ManagedOnlineDeploymentSchema(OnlineDeploymentSchema):
6970
)
7071
private_network_connection = ExperimentalField(fields.Bool())
7172
data_collector = NestedField(DataCollectorSchema)
73+
creation_context = NestedField(CreationContextSchema, dump_only=True)
7274

7375
@post_load
7476
def make(self, data: Any, **kwargs: Any) -> Any:

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -850,6 +850,7 @@ def _from_rest_object(cls, resource: RestOnlineDeploymentData) -> "ManagedOnline
850850
else None
851851
),
852852
provisioning_state=deployment.provisioning_state if hasattr(deployment, "provisioning_state") else None,
853+
creation_context=resource.system_data,
853854
)
854855

855856
def _merge_with(self, other: Any) -> None:

0 commit comments

Comments
 (0)