Airflow Summit 2025 is coming October 07-09. Register now for early bird ticket!

airflow.providers.amazon.aws.sensors.sagemaker_unified_studio

This module contains the Amazon SageMaker Unified Studio Notebook sensor.

Classes

SageMakerNotebookSensor

Waits for a Sagemaker Workflows Notebook execution to reach any of the status below.

Module Contents

class airflow.providers.amazon.aws.sensors.sagemaker_unified_studio.SageMakerNotebookSensor(*, execution_id, execution_name, **kwargs)[source]

Bases: airflow.providers.amazon.version_compat.BaseSensorOperator

Waits for a Sagemaker Workflows Notebook execution to reach any of the status below.

‘FAILED’, ‘STOPPED’, ‘COMPLETED’

Parameters:

  • execution_id (str) – The Sagemaker Workflows Notebook running execution identifier

  • execution_name (str) – The Sagemaker Workflows Notebook unique execution name

execution_id[source]
execution_name[source]
success_state = ['COMPLETED'][source]
in_progress_states = ['PENDING', 'RUNNING'][source]
hook()[source]
poke(context=None)[source]

Override when deriving this class.

execute(context)[source]

Derive when creating an operator.

The main method to execute the task. Context is the same dictionary used as when rendering jinja templates.

Refer to get_template_context for more context.

Was this entry helpful?