1 files changed, 28 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/spark_job.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/spark_job.py
new file mode 100644
index 00000000..f9363175
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/job/spark_job.py
@@ -0,0 +1,28 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+
+from azure.ai.ml._schema.core.fields import NestedField
+from azure.ai.ml._schema.job.identity import AMLTokenIdentitySchema, ManagedIdentitySchema, UserIdentitySchema
+from azure.ai.ml._schema.job.input_output_fields_provider import InputsField, OutputsField
+from azure.ai.ml._schema.spark_resource_configuration import SparkResourceConfigurationSchema
+from azure.ai.ml.constants import JobType
+
+from ..core.fields import ComputeField, StringTransformedEnum, UnionField
+from .base_job import BaseJobSchema
+from .parameterized_spark import ParameterizedSparkSchema
+
+
+class SparkJobSchema(ParameterizedSparkSchema, BaseJobSchema):
+    type = StringTransformedEnum(required=True, allowed_values=JobType.SPARK)
+    compute = ComputeField()
+    inputs = InputsField()
+    outputs = OutputsField()
+    resources = NestedField(SparkResourceConfigurationSchema)
+    identity = UnionField(
+        [
+            NestedField(ManagedIdentitySchema),
+            NestedField(AMLTokenIdentitySchema),
+            NestedField(UserIdentitySchema),
+        ]
+    )