Skip to content

Commit 9e80439

Browse files
authored
serverless for pipeline default compute (Azure#30086)
1 parent 0c79d2f commit 9e80439

File tree

1 file changed

+8
-3
lines changed
  • sdk/ml/azure-ai-ml/azure/ai/ml/_schema/pipeline

1 file changed

+8
-3
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/pipeline/settings.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,18 +6,23 @@
66

77
from marshmallow import INCLUDE, Schema, fields, post_dump, post_load
88

9-
from azure.ai.ml._schema.core.fields import ArmStr
9+
from azure.ai.ml._schema.core.fields import ArmStr, StringTransformedEnum, UnionField
1010
from azure.ai.ml._schema.pipeline.pipeline_component import NodeNameStr
1111
from azure.ai.ml._utils.utils import is_private_preview_enabled
12-
from azure.ai.ml.constants._common import AzureMLResourceType
12+
from azure.ai.ml.constants._common import AzureMLResourceType, SERVERLESS_COMPUTE
1313

1414

1515
class PipelineJobSettingsSchema(Schema):
1616
class Meta:
1717
unknown = INCLUDE
1818

1919
default_datastore = ArmStr(azureml_type=AzureMLResourceType.DATASTORE)
20-
default_compute = ArmStr(azureml_type=AzureMLResourceType.COMPUTE)
20+
default_compute = UnionField(
21+
[
22+
StringTransformedEnum(allowed_values=[SERVERLESS_COMPUTE]),
23+
ArmStr(azureml_type=AzureMLResourceType.COMPUTE),
24+
]
25+
)
2126
continue_on_step_failure = fields.Bool()
2227
force_rerun = fields.Bool()
2328

0 commit comments

Comments
 (0)