diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/monitoring/compute.py')
-rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/_schema/monitoring/compute.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/monitoring/compute.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/monitoring/compute.py new file mode 100644 index 00000000..483b4ac5 --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/monitoring/compute.py @@ -0,0 +1,23 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +# pylint: disable=unused-argument + +from marshmallow import fields, post_load +from azure.ai.ml._schema.core.schema import PatchedSchemaMeta + + +class ComputeConfigurationSchema(metaclass=PatchedSchemaMeta): + compute_type = fields.Str(allowed_values=["ServerlessSpark"]) + + +class ServerlessSparkComputeSchema(ComputeConfigurationSchema): + runtime_version = fields.Str() + instance_type = fields.Str() + + @post_load + def make(self, data, **kwargs): + from azure.ai.ml.entities._monitoring.compute import ServerlessSparkCompute + + return ServerlessSparkCompute(**data) |