aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py')
-rw-r--r--.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py48
1 files changed, 48 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py
new file mode 100644
index 00000000..669a96ad
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_deployment/deployment.py
@@ -0,0 +1,48 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+import logging
+
+from marshmallow import fields
+
+from azure.ai.ml._schema.assets.environment import AnonymousEnvironmentSchema, EnvironmentSchema
+from azure.ai.ml._schema.assets.model import AnonymousModelSchema
+from azure.ai.ml._schema.core.fields import ArmVersionedStr, NestedField, PathAwareSchema, RegistryStr, UnionField
+from azure.ai.ml.constants._common import AzureMLResourceType
+
+from .code_configuration_schema import CodeConfigurationSchema
+
+module_logger = logging.getLogger(__name__)
+
+
+class DeploymentSchema(PathAwareSchema):
+ name = fields.Str(required=True)
+ endpoint_name = fields.Str(required=True)
+ description = fields.Str(metadata={"description": "Description of the endpoint deployment."})
+ id = fields.Str()
+ tags = fields.Dict()
+ properties = fields.Dict()
+ model = UnionField(
+ [
+ RegistryStr(azureml_type=AzureMLResourceType.MODEL),
+ ArmVersionedStr(azureml_type=AzureMLResourceType.MODEL, allow_default_version=True),
+ NestedField(AnonymousModelSchema),
+ ],
+ metadata={"description": "Reference to the model asset for the endpoint deployment."},
+ )
+ code_configuration = NestedField(
+ CodeConfigurationSchema,
+ metadata={"description": "Code configuration for the endpoint deployment."},
+ )
+ environment = UnionField(
+ [
+ RegistryStr(azureml_type=AzureMLResourceType.ENVIRONMENT),
+ ArmVersionedStr(azureml_type=AzureMLResourceType.ENVIRONMENT, allow_default_version=True),
+ NestedField(EnvironmentSchema),
+ NestedField(AnonymousEnvironmentSchema),
+ ]
+ )
+ environment_variables = fields.Dict(
+ metadata={"description": "Environment variables configuration for the deployment."}
+ )