1 files changed, 44 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/schedule/schedule.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/schedule/schedule.py
new file mode 100644
index 00000000..fbde3e9b
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/schedule/schedule.py
@@ -0,0 +1,44 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from marshmallow import fields
+
+from azure.ai.ml._schema.core.fields import ArmStr, NestedField, UnionField
+from azure.ai.ml._schema.core.resource import ResourceSchema
+from azure.ai.ml._schema.job import CreationContextSchema
+from azure.ai.ml._schema.schedule.create_job import (
+    CommandCreateJobSchema,
+    CreateJobFileRefField,
+    PipelineCreateJobSchema,
+    SparkCreateJobSchema,
+)
+from azure.ai.ml._schema.schedule.trigger import CronTriggerSchema, RecurrenceTriggerSchema
+from azure.ai.ml.constants._common import AzureMLResourceType
+
+
+class ScheduleSchema(ResourceSchema):
+    name = fields.Str(attribute="name", required=True)
+    display_name = fields.Str(attribute="display_name")
+    trigger = UnionField(
+        [
+            NestedField(CronTriggerSchema),
+            NestedField(RecurrenceTriggerSchema),
+        ],
+    )
+    creation_context = NestedField(CreationContextSchema, dump_only=True)
+    is_enabled = fields.Boolean(dump_only=True)
+    provisioning_state = fields.Str(dump_only=True)
+    properties = fields.Dict(keys=fields.Str(), values=fields.Str(allow_none=True))
+
+
+class JobScheduleSchema(ScheduleSchema):
+    create_job = UnionField(
+        [
+            ArmStr(azureml_type=AzureMLResourceType.JOB),
+            CreateJobFileRefField,
+            NestedField(PipelineCreateJobSchema),
+            NestedField(CommandCreateJobSchema),
+            NestedField(SparkCreateJobSchema),
+        ]
+    )