1 files changed, 60 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/__init__.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/__init__.py
new file mode 100644
index 00000000..115a65bb
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/__init__.py
@@ -0,0 +1,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",
+]