aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py
diff options
context:
space:
mode:
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py')
-rw-r--r--.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py41
1 files changed, 41 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py
new file mode 100644
index 00000000..1c011bc9
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/parameterized_command.py
@@ -0,0 +1,41 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from marshmallow import fields
+
+from azure.ai.ml._schema.core.fields import (
+ CodeField,
+ DistributionField,
+ EnvironmentField,
+ ExperimentalField,
+ NestedField,
+)
+from azure.ai.ml._schema.core.schema import PathAwareSchema
+from azure.ai.ml._schema.job.input_output_entry import InputLiteralValueSchema
+from azure.ai.ml._schema.job_resource_configuration import JobResourceConfigurationSchema
+from azure.ai.ml._schema.queue_settings import QueueSettingsSchema
+
+from ..core.fields import UnionField
+
+
+class ParameterizedCommandSchema(PathAwareSchema):
+ command = fields.Str(
+ metadata={
+ # pylint: disable=line-too-long
+ "description": "The command run and the parameters passed. This string may contain place holders of inputs in {}. "
+ },
+ required=True,
+ )
+ code = CodeField()
+ environment = EnvironmentField(required=True)
+ environment_variables = UnionField(
+ [
+ fields.Dict(keys=fields.Str(), values=fields.Str()),
+ # Used for binding environment variables
+ NestedField(InputLiteralValueSchema),
+ ]
+ )
+ resources = NestedField(JobResourceConfigurationSchema)
+ distribution = DistributionField()
+ queue_settings = ExperimentalField(NestedField(QueueSettingsSchema))