1 files changed, 25 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/data.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/data.py
new file mode 100644
index 00000000..e14afd9b
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/data.py
@@ -0,0 +1,25 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from marshmallow import fields, validate
+
+from azure.ai.ml.constants._common import AssetTypes
+
+from .artifact import ArtifactSchema
+from .asset import AnonymousAssetSchema
+
+
+class DataSchema(ArtifactSchema):
+    path = fields.Str(metadata={"description": "URI pointing to a file or folder."}, required=True)
+    properties = fields.Dict(dump_only=True)
+    type = fields.Str(
+        metadata={"description": "the type of data. Valid values are uri_file, uri_folder, or mltable."},
+        validate=validate.OneOf([AssetTypes.URI_FILE, AssetTypes.URI_FOLDER, AssetTypes.MLTABLE]),
+        dump_default=AssetTypes.URI_FOLDER,
+        error_messages={"validator_failed": "value must be uri_file, uri_folder, or mltable."},
+    )
+
+
+class AnonymousDataSchema(DataSchema, AnonymousAssetSchema):
+    pass