Skip to content

RetryConfig

kedro_azureml_pipeline.config.RetryConfig

Bases: BaseModel

Retry settings for Azure ML pipeline steps.

Maps to azure.ai.ml.entities.RetrySettings applied on each invoked command component.

Parameters

Name Type Description Default
max_retries int

Maximum number of retry attempts for failed steps.

 required
timeout int or None

Per-attempt timeout in seconds, or None for no limit.

 required

See Also

JobConfig : Uses retry settings per job.

Source Code

View on GitHub

Show/Hide source 
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
class RetryConfig(BaseModel):
    """Retry settings for Azure ML pipeline steps.

    Maps to ``azure.ai.ml.entities.RetrySettings`` applied on each
    invoked command component.

    Parameters
    ----------
    max_retries : int
        Maximum number of retry attempts for failed steps.
    timeout : int or None
        Per-attempt timeout in seconds, or ``None`` for no limit.

    See Also
    --------
    [JobConfig][kedro_azureml_pipeline.config.JobConfig] : Uses retry settings per job.
    """

    model_config = ConfigDict(extra="forbid")

    max_retries: int = Field(ge=1, description="Maximum number of retry attempts for failed steps.")
    timeout: int | None = Field(default=None, ge=1, description="Per-attempt timeout in seconds, or None for no limit.")