diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/automl/automl_job.py')
| -rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/_schema/automl/automl_job.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/automl/automl_job.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/automl/automl_job.py new file mode 100644 index 00000000..ebec82c7 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/automl/automl_job.py @@ -0,0 +1,21 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +from marshmallow import fields + +from azure.ai.ml._schema.core.fields import ExperimentalField, NestedField, StringTransformedEnum +from azure.ai.ml._schema.job import BaseJobSchema +from azure.ai.ml._schema.job.input_output_fields_provider import OutputsField +from azure.ai.ml._schema.job_resource_configuration import JobResourceConfigurationSchema +from azure.ai.ml._schema.queue_settings import QueueSettingsSchema +from azure.ai.ml.constants import JobType + + +class AutoMLJobSchema(BaseJobSchema): + type = StringTransformedEnum(required=True, allowed_values=JobType.AUTOML) + environment_id = fields.Str() + environment_variables = fields.Dict(keys=fields.Str(), values=fields.Str()) + outputs = OutputsField() + resources = NestedField(JobResourceConfigurationSchema()) + queue_settings = ExperimentalField(NestedField(QueueSettingsSchema)) |