Getting Started

Installation

pip install capo-sagemakerjobruntime

Usage

from capo_sagemakerjobruntime import AsyncSagemakerJobRuntimeClient


async def main():
    async with AsyncSagemakerJobRuntimeClient() as sagemaker_job_runtime:
        # Example: call the complete_rollout operation
        response = await sagemaker_job_runtime.complete_rollout()
        print(response)

Streaming Response

Some operations return a streaming response body. Use the operation as an async context manager and iterate over the response field to read chunks.

from capo_sagemakerjobruntime import AsyncSagemakerJobRuntimeClient


async def main():
    async with AsyncSagemakerJobRuntimeClient() as sagemaker_job_runtime:
        # Example: call sample_with_response_stream and read the streaming response
        async with sagemaker_job_runtime.sample_with_response_stream() as response:
            async for chunk in response["body"]:
                print(chunk)

Error Handling

The SDK raises exceptions for errors returned by the API. Catch them to handle failures gracefully.

from capo_sagemakerjobruntime import AsyncSagemakerJobRuntimeClient
from capo_sagemakerjobruntime.error import AccessDeniedException


async def main():
    async with AsyncSagemakerJobRuntimeClient() as sagemaker_job_runtime:
        try:
            await sagemaker_job_runtime.complete_rollout()
        except AccessDeniedException as e:
            print(f"Error: {e}")
            print(e.data)  # additional error data

Retrying

The SDK retries failed operations automatically. Retry behaviour follows the Smithy specification: errors are retried based on their is_retryable and is_throttling_error attributes. Throttling errors use a longer base delay. Network-level failures (connection errors and timeouts) are also retried. Non-retryable errors, such as client errors without the @retryable trait, are raised immediately without further attempts.

The number of attempts defaults to 3 and can be changed at the client level via retry_max_attempts, or per call via config_overrides.

from capo_sagemakerjobruntime import AsyncSagemakerJobRuntimeClient


async def main():
    async with AsyncSagemakerJobRuntimeClient() as sagemaker_job_runtime:
        # Default: 3 attempts for every operation
        response = await sagemaker_job_runtime.complete_rollout()

        # Override per operation
        response = await sagemaker_job_runtime.complete_rollout(config_overrides={"retry_max_attempts": 5})

        # Disable retries for this call
        response = await sagemaker_job_runtime.complete_rollout(config_overrides={"retry_max_attempts": 1})