aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/compute/kubernetes_compute.py
blob: a84102cac825e0bf2dcbfdc9112c2d997e04af5f (about) (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------
from marshmallow import fields

from azure.ai.ml.constants._compute import ComputeType

from ..core.fields import NestedField, StringTransformedEnum
from .compute import ComputeSchema, IdentitySchema


class KubernetesComputeSchema(ComputeSchema):
    type = StringTransformedEnum(allowed_values=[ComputeType.KUBERNETES], required=True)
    namespace = fields.Str(required=True, dump_default="default")
    properties = fields.Dict()
    identity = NestedField(IdentitySchema)