1 files changed, 45 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/job_limits.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/job_limits.py
new file mode 100644
index 00000000..850e9b3d
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/job_limits.py
@@ -0,0 +1,45 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+# pylint: disable=unused-argument
+
+from marshmallow import fields, post_load, validate
+
+from azure.ai.ml._schema.core.schema_meta import PatchedSchemaMeta
+
+
+class CommandJobLimitsSchema(metaclass=PatchedSchemaMeta):
+    timeout = fields.Int()
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import CommandJobLimits
+
+        return CommandJobLimits(**data)
+
+
+class SweepJobLimitsSchema(metaclass=PatchedSchemaMeta):
+    max_concurrent_trials = fields.Int(metadata={"description": "Sweep Job max concurrent trials."})
+    max_total_trials = fields.Int(
+        metadata={"description": "Sweep Job max total trials."},
+        required=True,
+    )
+    timeout = fields.Int(
+        metadata={"description": "The max run duration in Seconds, after which the job will be cancelled."}
+    )
+    trial_timeout = fields.Int(metadata={"description": "Sweep Job Trial timeout value."})
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.sweep import SweepJobLimits
+
+        return SweepJobLimits(**data)
+
+
+class DoWhileLimitsSchema(metaclass=PatchedSchemaMeta):
+    max_iteration_count = fields.Int(
+        metadata={"description": "The max iteration for do_while loop."},
+        validate=validate.Range(min=1, max=1000),
+        required=True,
+    )