2 files changed, 35 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/__init__.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/__init__.py
new file mode 100644
index 00000000..b95c2d6d
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/__init__.py
@@ -0,0 +1,11 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)
+
+from .notification_schema import NotificationSchema
+
+__all__ = [
+    "NotificationSchema",
+]
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/notification_schema.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/notification_schema.py
new file mode 100644
index 00000000..21245bc9
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/_notification/notification_schema.py
@@ -0,0 +1,24 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+from marshmallow import fields, validate, post_load
+
+from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
+
+
+class NotificationSchema(metaclass=PatchedSchemaMeta):
+    email_on = fields.List(
+        fields.Str(validate=validate.OneOf(["JobCompleted", "JobFailed", "JobCancelled"])),
+        required=True,
+        allow_none=False,
+    )
+    emails = fields.List(fields.Str, required=True, allow_none=False)
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities._notification.notification import Notification
+
+        return Notification(**data)