about summary refs log tree commit diff
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))