1 files changed, 38 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job_resource_configuration.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job_resource_configuration.py
new file mode 100644
index 00000000..859eef31
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job_resource_configuration.py
@@ -0,0 +1,38 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from marshmallow import fields, post_load
+
+from azure.ai.ml._schema.core.fields import UnionField
+
+from .resource_configuration import ResourceConfigurationSchema
+
+
+class JobResourceConfigurationSchema(ResourceConfigurationSchema):
+    locations = fields.List(fields.Str())
+    shm_size = fields.Str(
+        metadata={
+            "description": (
+                "The size of the docker container's shared memory block. "
+                "This should be in the format of `<number><unit>` where number as "
+                "to be greater than 0 and the unit can be one of "
+                "`b` (bytes), `k` (kilobytes), `m` (megabytes), or `g` (gigabytes)."
+            )
+        }
+    )
+    max_instance_count = fields.Int(
+        metadata={"description": "The maximum number of instances to make available to this job."}
+    )
+    docker_args = UnionField(
+        [
+            fields.Str(metadata={"description": "arguments to pass to the Docker run command."}),
+            fields.List(fields.Str()),
+        ]
+    )
+
+    @post_load
+    def make(self, data, **kwargs):
+        from azure.ai.ml.entities import JobResourceConfiguration
+
+        return JobResourceConfiguration(**data)