1 files changed, 24 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/artifact.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/artifact.py
new file mode 100644
index 00000000..fc107a78
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/assets/artifact.py
@@ -0,0 +1,24 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+import logging
+
+from marshmallow import fields, post_load
+
+from azure.ai.ml.constants._common import BASE_PATH_CONTEXT_KEY
+
+from .asset import AssetSchema
+
+module_logger = logging.getLogger(__name__)
+
+
+class ArtifactSchema(AssetSchema):
+    datastore = fields.Str(metadata={"description": "Name of the datastore to upload to."}, required=False)
+
+    @post_load
+    def make(self, data, **kwargs):
+        data[BASE_PATH_CONTEXT_KEY] = self.context[BASE_PATH_CONTEXT_KEY]
+        return data