blob: 859eef31675c9f3eab4755d23557efa121f0c215 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------
from marshmallow import fields, post_load
from azure.ai.ml._schema.core.fields import UnionField
from .resource_configuration import ResourceConfigurationSchema
class JobResourceConfigurationSchema(ResourceConfigurationSchema):
locations = fields.List(fields.Str())
shm_size = fields.Str(
metadata={
"description": (
"The size of the docker container's shared memory block. "
"This should be in the format of `<number><unit>` where number as "
"to be greater than 0 and the unit can be one of "
"`b` (bytes), `k` (kilobytes), `m` (megabytes), or `g` (gigabytes)."
)
}
)
max_instance_count = fields.Int(
metadata={"description": "The maximum number of instances to make available to this job."}
)
docker_args = UnionField(
[
fields.Str(metadata={"description": "arguments to pass to the Docker run command."}),
fields.List(fields.Str()),
]
)
@post_load
def make(self, data, **kwargs):
from azure.ai.ml.entities import JobResourceConfiguration
return JobResourceConfiguration(**data)
|
