Retry strategy

The hera.workflows.retry_strategy module provides retry strategy functionality, along with necessary dependencies such as retry policy.

RetryPolicy

An enum that holds options for retry policy.

Source code in src/hera/workflows/retry_strategy.py

class RetryPolicy(Enum):
    """An enum that holds options for retry policy."""

    always = "Always"
    """Retry all failed steps"""

    on_failure = "OnFailure"
    """Retry steps whose main container is marked as failed in Kubernetes"""

    on_error = "OnError"
    """Retry steps that encounter Argo controller errors, or whose init or wait containers fail"""

    on_transient_error = "OnTransientError"
    """Retry steps that encounter errors defined as transient, or errors matching the `TRANSIENT_ERROR_PATTERN`
    environment variable.
    Available in version 3.0 and later.
    """

    def __str__(self) -> str:
        """Assembles the `value` representation of the enum as a string."""
        return str(self.value)

always

always = 'Always'

Retry all failed steps

on_error

on_error = 'OnError'

Retry steps that encounter Argo controller errors, or whose init or wait containers fail

on_failure

on_failure = 'OnFailure'

Retry steps whose main container is marked as failed in Kubernetes

on_transient_error

on_transient_error = 'OnTransientError'

Retry steps that encounter errors defined as transient, or errors matching the TRANSIENT_ERROR_PATTERN environment variable. Available in version 3.0 and later.

RetryStrategy

RetryStrategy configures how an Argo job should retry.

Source code in src/hera/workflows/retry_strategy.py

@dataclass(kw_only=True)
class RetryStrategy:
    """`RetryStrategy` configures how an Argo job should retry."""

    affinity: Optional[RetryAffinity] = None
    """affinity dictates the affinity of the retried jobs"""

    backoff: Optional[Backoff] = None
    """backoff dictates how long should a job wait for before retrying"""

    expression: Optional[str] = None
    """the expression field supports the expression of complex rules regarding retry behavior"""

    limit: Optional[Union[int, str, IntOrString]] = None
    """the hard numeric limit of how many times a jobs should retry"""

    retry_policy: Optional[Union[str, RetryPolicy]] = None
    """the policy dictates, at a high level, under what conditions should a job retry"""

    def build(self) -> _ModelRetryStrategy:
        """Builds the generated `RetryStrategy` representation of the retry strategy."""
        limit = IntOrString(root=self.limit) if isinstance(self.limit, (str, int)) else self.limit
        return _ModelRetryStrategy(
            affinity=self.affinity,
            backoff=self.backoff,
            expression=self.expression,
            limit=limit,
            retry_policy=str(self.retry_policy) if self.retry_policy is not None else None,
        )

affinity

affinity: Optional[RetryAffinity] = None

affinity dictates the affinity of the retried jobs

backoff

backoff: Optional[Backoff] = None

backoff dictates how long should a job wait for before retrying

expression

expression: Optional[str] = None

the expression field supports the expression of complex rules regarding retry behavior

limit

limit: Optional[Union[int, str, IntOrString]] = None

the hard numeric limit of how many times a jobs should retry

retry_policy

retry_policy: Optional[Union[str, RetryPolicy]] = None

the policy dictates, at a high level, under what conditions should a job retry

build

build() -> RetryStrategy

Builds the generated RetryStrategy representation of the retry strategy.

Source code in src/hera/workflows/retry_strategy.py

def build(self) -> _ModelRetryStrategy:
    """Builds the generated `RetryStrategy` representation of the retry strategy."""
    limit = IntOrString(root=self.limit) if isinstance(self.limit, (str, int)) else self.limit
    return _ModelRetryStrategy(
        affinity=self.affinity,
        backoff=self.backoff,
        expression=self.expression,
        limit=limit,
        retry_policy=str(self.retry_policy) if self.retry_policy is not None else None,
    )

Comments