blob: ebec82c75434740f0123fa725114196b9da86738 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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))
|
