airflow.providers.amazon.aws.hooks.sagemaker_unified_studio¶

This module contains the Amazon SageMaker Unified Studio Notebook hook.

Classes¶

SageMakerNotebookHook

Interact with Sagemaker Unified Studio Workflows for executing Jupyter notebooks, querybooks, and visual ETL jobs.

Module Contents¶

class airflow.providers.amazon.aws.hooks.sagemaker_unified_studio.SageMakerNotebookHook(execution_name, input_config=None, domain_id=None, project_id=None, output_config=None, domain_region=None, compute=None, termination_condition=None, tags=None, waiter_delay=10, waiter_max_attempts=1440, *args, **kwargs)[source]¶

Bases: airflow.providers.common.compat.sdk.BaseHook

Interact with Sagemaker Unified Studio Workflows for executing Jupyter notebooks, querybooks, and visual ETL jobs.

This hook provides a wrapper around the Sagemaker Workflows Notebook Execution API.

execution_name[source]¶

domain_id = None[source]¶

project_id = None[source]¶

domain_region = None[source]¶

input_config[source]¶

output_config[source]¶

compute = None[source]¶

termination_condition[source]¶

tags[source]¶

waiter_delay = 10[source]¶

waiter_max_attempts = 1440[source]¶

start_notebook_execution()[source]¶

get_notebook_execution(execution_id)[source]¶

Fetch the status of a SageMaker Notebook Job execution.

wait_for_execution_completion(execution_id, context)[source]¶