기계 번역으로 제공되는 번역입니다. 제공된 번역과 원본 영어의 내용이 상충하는 경우에는 영어 버전이 우선합니다.

파이프라인 실행 예약

Amazon 를 사용하여 Amazon EventBridge SageMaker Pipelines 실행을 예약할 수 있습니다. Amazon SageMaker Pipelines은 Amazon EventBridge에서 대상으로 지원됩니다. 이를 통해 이벤트 버스의 모든 이벤트를 기반으로 모델 구축 파이프라인의 실행을 시작할 수 있습니다. 를 사용하면 파이프라인 실행을 자동화하고 훈련 작업 또는 엔드포인트 상태 변경과 같은 이벤트에 자동으로 대응할 EventBridge수 있습니다. 이벤트에는 Amazon S3 버킷에 업로드되는 새 파일, 드리프트로 인한 Amazon SageMaker 엔드포인트 상태 변경, Amazon Simple Notification Service(SNS) 주제가 포함됩니다.

다음 파이프라인 작업을 자동으로 시작할 수 있습니다. 

SageMaker 작업 예약에 대한 자세한 내용은 SageMaker Amazon으로 자동화를 참조하세요 EventBridge.

Amazon에서 파이프라인 예약 EventBridge

Amazon CloudWatch Events로 파이프라인 실행을 시작하려면 EventBridge애널 을 생성해야 합니다. 이벤트에 대한 규칙을 생성할 때 가 규칙과 일치하는 이벤트를 EventBridge 수신할 때 수행할 대상 작업을 지정합니다. 이벤트가 규칙과 일치하면 는 이벤트를 지정된 대상으로 EventBridge 보내고 규칙에 정의된 작업을 시작합니다.

다음 자습서에서는 콘솔 또는 를 EventBridge 사용하여 파이프라인 실행을 EventBridge 예약하는 방법을 보여줍니다 AWS CLI. 

사전 조건

콘솔을 사용하여 EventBridge EventBridge 규칙 생성

다음 절차에서는 EventBridge 콘솔을 사용하여 EventBridge 규칙을 생성하는 방법을 보여줍니다. 

이제 규칙이 적용되어 파이프라인 실행을 시작할 준비가 되었습니다.

를 사용하여 EventBridge 규칙 생성 AWS CLI

다음 절차에서는 를 사용하여 EventBridge 규칙을 생성하는 방법을 보여줍니다 AWS CLI.

SageMaker Python으로 파이프라인 예약 SDK

다음 섹션에서는 SageMaker Python 을 사용하여 EventBridge 리소스에 액세스하고 파이프라인 일정을 생성할 수 있는 권한을 설정하는 방법을 보여줍니다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>"])