airflow.providers.amazon.aws.triggers.sagemaker_unified_studio¶

此模块包含 Amazon SageMaker Unified Studio Notebook 作业触发器。

类¶

SageMakerNotebookJobTrigger

监视 notebook 作业，并在其完成后触发。

模块内容¶

class airflow.providers.amazon.aws.triggers.sagemaker_unified_studio.SageMakerNotebookJobTrigger(execution_id, execution_name, waiter_delay, waiter_max_attempts, **kwargs)[源码]¶

基类: airflow.triggers.base.BaseTrigger

监视 notebook 作业，并在其完成后触发。

示例

from airflow.providers.amazon.aws.triggers.sagemaker_unified_studio import SageMakerNotebookJobTrigger

notebook_trigger = SageMakerNotebookJobTrigger(
    execution_id="notebook_job_1234",
    execution_name="notebook_task",
    waiter_delay=10,
    waiter_max_attempts=1440,
)

参数:

execution_id – 任务的唯一且有意义的 ID。
execution_name – 任务的唯一且有意义的名称。
waiter_delay – 检查 notebook 执行状态的间隔（秒）。
waiter_max_attempts – 返回 FAILED 之前的等待尝试次数。

execution_id[源码]¶

execution_name[源码]¶

waiter_delay[源码]¶

waiter_max_attempts[源码]¶

serialize()[源码]¶

返回重构此触发器所需的信息。

返回:: 包含 (类路径, 重新实例化所需的关键字参数) 的元组。

async run()[源码]¶

在异步上下文中运行触发器。

当触发器想要触发事件时，应生成一个 Event；如果完成，则返回 None。因此，单事件触发器应生成事件后立即返回。

如果它生成事件，很可能会很快恢复，但也可能不会（例如，如果工作负载被移动到另一个触发器进程，或者多事件触发器被用于单事件任务延迟）。

无论哪种情况，Trigger 类都应假定它们会被持久化，并在不再需要时依赖于 cleanup() 函数的调用。