1 files changed, 23 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/command_job.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/command_job.py
new file mode 100644
index 00000000..9cce7de7
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/command_job.py
@@ -0,0 +1,23 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from marshmallow import fields
+
+from azure.ai.ml._schema.core.fields import NestedField, StringTransformedEnum
+from azure.ai.ml._schema.job.input_output_fields_provider import InputsField, OutputsField
+from azure.ai.ml.constants import JobType
+
+from .base_job import BaseJobSchema
+from .job_limits import CommandJobLimitsSchema
+from .parameterized_command import ParameterizedCommandSchema
+
+
+class CommandJobSchema(ParameterizedCommandSchema, BaseJobSchema):
+    type = StringTransformedEnum(allowed_values=JobType.COMMAND)
+    # do not promote it as CommandComponent has no field named 'limits'
+    limits = NestedField(CommandJobLimitsSchema)
+    parameters = fields.Dict(dump_only=True)
+    inputs = InputsField()
+    outputs = OutputsField()
+    parent_job_name = fields.Str()