本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

排程管道執行

您可以使用 Amazon EventBridge 排程 Amazon SageMaker 管道執行。 EventBridge Amazon EventBridge 中的目標支援 Amazon SageMaker 管道。 EventBridge 這可讓您根據事件匯流排中的任何事件，啟動模型建置管道執行。使用 EventBridge，您可以自動執行管道，並自動回應訓練工作或端點狀態變更等事件。事件包括正在上傳至 Amazon S3 儲存貯體的新檔案、因偏離而變更 Amazon SageMaker AI 端點的狀態，以及 Amazon Simple Notification Service (SNS) 主題。

可自動啟動下列管道動作： 

如需排程 SageMaker AI 任務的詳細資訊，請參閱使用 Amazon EventBridge 自動化 SageMaker AI。

使用 Amazon EventBridge 為管道排程

若要使用 Amazon CloudWatch Events 開始管道執行，您必須建立 EventBridge 規則。當您為事件建立規則時，您可以指定當 EventBridge 收到符合規則的事件時要採取的目標動作。當事件符合規則時，EventBridge 會將事件傳送到指定目標並啟動規則中定義的動作。

下列教學課程示範如何使用 EventBridge 主控台或 AWS CLI為管道執行排程。 

先決條件

使用 EventBridge 主控台建立 EventBridge 規則

下列程序示範如何使用 EventBridge 主控台建立 EventBridge 規則。 

您的規則現已生效，可用於啟動管道執行了。

使用 AWS CLI 建立 EventBridge 規則

下列程序示範如何使用 AWS CLI建立 EventBridge 規則。

使用 SageMaker Python SDK 排程管道

下列各節說明如何設定存取 EventBridge 資源的許可，並使用 SageMaker Python SDK 建立管道排程。

所需的許可

您需要具有使用管道排程器的必要許可。完成下列步驟以設定您的許可：

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Effect": "Allow",
            "Principal": {
                "Service": [
                    "scheduler.amazonaws.com",
                    "sagemaker.amazonaws.com"
                ]
            },
        "Action": "sts:AssumeRole"
        }
    ]
}

建立管道排程

使用PipelineSchedule建構函數，您可以將管道排程為執行一次或按預定間隔執行。管道排程的類型必須為 at、 rate或 cron。這組排程類型是 EventBridge 排程選項的延伸。如需如何使用 PipelineSchedule類別的詳細資訊，請參閱 sagemaker.workflow.triggers.PipelineSchedule。下列範例示範如何使用 建立每個排程類型PipelineSchedule。

from sagemaker.workflow.triggers import PipelineSchedule

# schedules a pipeline run for 12/13/2023 at time 10:15:20 UTC
my_datetime_schedule = PipelineSchedule(
    name="<schedule-name>", 
    at=datetime(2023, 12, 13, 10, 15, 20)
)

# schedules a pipeline run every 5 minutes
my_rate_schedule = PipelineSchedule(
    name="<schedule-name>", 
    rate=(5, "minutes")
)

# schedules a pipeline run at 10:15am UTC on the last Friday of each month during the years 2022 to 2023
my_cron_schedule = PipelineSchedule(
    name="<schedule-name>", 
    cron="15 10 ? * 6L 2022-2023"
)

將觸發條件連接至您的管道

若要將 PipelineSchedule 連接到您的管道，請使用觸發條件清單，在建立的管道物件上叫用 put_triggers呼叫。如果您收到回應 ARN，則已成功在帳戶中建立排程，EventBridge 會在指定的時間或速率開始調用目標管道。您必須指定具有正確許可的角色，才能將觸發條件連接至父管道。如果您未提供，管道會擷取用於從組態檔案建立管道的預設角色。

下列範例示範如何將排程連接至管道。

scheduled_pipeline = Pipeline(
    name="<pipeline-name>",
    steps=[...],
    sagemaker_session=<sagemaker-session>,
)
custom_schedule = PipelineSchedule(
    name="<schedule-name>", 
    at=datetime(year=2023, month=12, date=25, hour=10, minute=30, second=30)
)
scheduled_pipeline.put_triggers(triggers=[custom_schedule], role_arn=<role>)

描述目前的觸發條件

若要擷取所建立管道觸發的相關資訊，您可以使用觸發名稱叫用 describe_trigger() API。此命令會傳回所建立排程表達式的詳細資訊，例如其開始時間、啟用狀態和其他實用資訊。下列程式碼片段顯示範例調用：

scheduled_pipeline.describe_trigger(name="<schedule-name>")

清除觸發程序資源

刪除管道之前，請先清除現有的觸發條件，以避免帳戶中的資源洩漏。您應該先刪除觸發條件，再銷毀父管道。您可以透過將觸發條件名稱清單傳遞至 delete_triggers API 來刪除觸發條件。下列程式碼片段示範如何刪除觸發。

pipeline.delete_triggers(trigger_names=["<schedule-name>"])