Skip to content

Commit b25a9f1

Browse files
Fix final marshmallow 4.x compatibility issue: Replace fields.Str with allowed_values parameter
Co-authored-by: kshitij-microsoft <[email protected]>
1 parent ce15d4b commit b25a9f1

File tree

1 file changed

+2
-1
lines changed
  • sdk/ml/azure-ai-ml/azure/ai/ml/_schema/monitoring

1 file changed

+2
-1
lines changed

sdk/ml/azure-ai-ml/azure/ai/ml/_schema/monitoring/compute.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,11 @@
66

77
from marshmallow import fields, post_load
88
from azure.ai.ml._schema.core.schema import PatchedSchemaMeta
9+
from azure.ai.ml._schema.core.fields import StringTransformedEnum
910

1011

1112
class ComputeConfigurationSchema(metaclass=PatchedSchemaMeta):
12-
compute_type = fields.Str(allowed_values=["ServerlessSpark"])
13+
compute_type = StringTransformedEnum(allowed_values=["ServerlessSpark"])
1314

1415

1516
class ServerlessSparkComputeSchema(ComputeConfigurationSchema):

0 commit comments

Comments
 (0)