diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/custom_applications.py')
-rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/custom_applications.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/custom_applications.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/custom_applications.py new file mode 100644 index 00000000..66fa587c --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/custom_applications.py @@ -0,0 +1,60 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +# pylint: disable=unused-argument +from marshmallow import fields +from marshmallow.decorators import post_load + +from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum +from azure.ai.ml._schema.core.schema_meta import PatchedSchemaMeta +from azure.ai.ml.constants._compute import CustomApplicationDefaults + + +class ImageSettingsSchema(metaclass=PatchedSchemaMeta): + reference = fields.Str() + + @post_load + def make(self, data, **kwargs): + from azure.ai.ml.entities._compute._custom_applications import ImageSettings + + return ImageSettings(**data) + + +class EndpointsSettingsSchema(metaclass=PatchedSchemaMeta): + target = fields.Int() + published = fields.Int() + + @post_load + def make(self, data, **kwargs): + from azure.ai.ml.entities._compute._custom_applications import EndpointsSettings + + return EndpointsSettings(**data) + + +class VolumeSettingsSchema(metaclass=PatchedSchemaMeta): + source = fields.Str() + target = fields.Str() + + @post_load + def make(self, data, **kwargs): + from azure.ai.ml.entities._compute._custom_applications import VolumeSettings + + return VolumeSettings(**data) + + +class CustomApplicationsSchema(metaclass=PatchedSchemaMeta): + name = fields.Str(required=True) + type = StringTransformedEnum(allowed_values=[CustomApplicationDefaults.DOCKER]) + image = NestedField(ImageSettingsSchema) + endpoints = fields.List(NestedField(EndpointsSettingsSchema)) + environment_variables = fields.Dict() + bind_mounts = fields.List(NestedField(VolumeSettingsSchema)) + + @post_load + def make(self, data, **kwargs): + from azure.ai.ml.entities._compute._custom_applications import ( + CustomApplications, + ) + + return CustomApplications(**data) |