diff options
Diffstat (limited to '.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/component/retry_settings.py')
| -rw-r--r-- | .venv/lib/python3.12/site-packages/azure/ai/ml/_schema/component/retry_settings.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/component/retry_settings.py b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/component/retry_settings.py new file mode 100644 index 00000000..bac2c54d --- /dev/null +++ b/.venv/lib/python3.12/site-packages/azure/ai/ml/_schema/component/retry_settings.py @@ -0,0 +1,13 @@ +# --------------------------------------------------------- +# Copyright (c) Microsoft Corporation. All rights reserved. +# --------------------------------------------------------- + +from marshmallow import fields + +from azure.ai.ml._schema.core.schema import PatchedSchemaMeta +from azure.ai.ml._schema.core.fields import DataBindingStr, UnionField + + +class RetrySettingsSchema(metaclass=PatchedSchemaMeta): + timeout = UnionField([fields.Int(), DataBindingStr]) + max_retries = UnionField([fields.Int(), DataBindingStr]) |