airflow.providers.amazon.aws.triggers.sagemaker_unified_studio_notebook¶

Trigger for monitoring SageMaker Unified Studio Notebook runs asynchronously.

Attributes¶

TWELVE_HOURS_IN_MINUTES

Classes¶

SageMakerUnifiedStudioNotebookTrigger

Watches an asynchronous notebook run, triggering when it reaches a terminal state.

Module Contents¶

airflow.providers.amazon.aws.triggers.sagemaker_unified_studio_notebook.TWELVE_HOURS_IN_MINUTES = 720[source]¶

class airflow.providers.amazon.aws.triggers.sagemaker_unified_studio_notebook.SageMakerUnifiedStudioNotebookTrigger(notebook_run_id, domain_identifier, owning_project_identifier, waiter_delay=10, timeout_configuration=None, aws_conn_id=None, **kwargs)[source]¶

Bases: airflow.providers.amazon.aws.triggers.base.AwsBaseWaiterTrigger

Watches an asynchronous notebook run, triggering when it reaches a terminal state.

Uses a custom boto waiter (notebook_run_complete) defined in waiters/datazone.json to poll the DataZone GetNotebookRun API.

Parameters:

notebook_run_id (str) – The ID of the notebook run to monitor.
domain_identifier (str) – The ID of the DataZone domain.
owning_project_identifier (str) – The ID of the DataZone project.
waiter_delay (int) – Interval in seconds between polls (default: 10).
waiter_max_attempts – Maximum number of poll attempts.
aws_conn_id (str | None) – The Airflow connection used for AWS credentials.
timeout_configuration (dict | None) – Optional timeout settings. When provided, the maximum number of poll attempts is derived from runTimeoutInMinutes * 60 / waiter_delay. Defaults to a 12-hour timeout when omitted. Example: {“runTimeoutInMinutes”: 720}

notebook_run_id[source]¶

domain_identifier[source]¶

owning_project_identifier[source]¶

timeout_configuration = None[source]¶

hook()[source]¶

Override in subclasses to return the right hook.