aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/__init__.py
blob: 115a65bbe7099922006263ca87d5364f63455693 (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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------

__path__ = __import__("pkgutil").extend_path(__path__, __name__)

from ._data_import import DataImportSchema
from ._sweep import SweepJobSchema
from .assets.code_asset import AnonymousCodeAssetSchema, CodeAssetSchema
from .assets.data import DataSchema
from .assets.environment import AnonymousEnvironmentSchema, EnvironmentSchema
from .assets.index import IndexAssetSchema
from .assets.model import ModelSchema
from .assets.workspace_asset_reference import WorkspaceAssetReferenceSchema
from .component import CommandComponentSchema
from .core.fields import (
    ArmStr,
    ArmVersionedStr,
    ExperimentalField,
    NestedField,
    RegistryStr,
    StringTransformedEnum,
    UnionField,
)
from .core.schema import PathAwareSchema, YamlFileSchema
from .core.schema_meta import PatchedSchemaMeta
from .job import CommandJobSchema, ParallelJobSchema, SparkJobSchema

# TODO: enable in PuP
# from .job import ImportJobSchema
# from .component import ImportComponentSchema

__all__ = [
    # "ImportJobSchema",
    # "ImportComponentSchema",
    "ArmStr",
    "ArmVersionedStr",
    "DataSchema",
    "StringTransformedEnum",
    "CodeAssetSchema",
    "CommandJobSchema",
    "SparkJobSchema",
    "ParallelJobSchema",
    "EnvironmentSchema",
    "AnonymousEnvironmentSchema",
    "NestedField",
    "PatchedSchemaMeta",
    "PathAwareSchema",
    "ModelSchema",
    "SweepJobSchema",
    "UnionField",
    "YamlFileSchema",
    "CommandComponentSchema",
    "AnonymousCodeAssetSchema",
    "ExperimentalField",
    "RegistryStr",
    "WorkspaceAssetReferenceSchema",
    "DataImportSchema",
    "IndexAssetSchema",
]