Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Planifier les exécutions du pipeline

Vous pouvez planifier vos exécutions Amazon SageMaker Pipelines à l'aide d'Amazon EventBridge. Amazon SageMaker Pipelines est pris en charge en tant que cible dans Amazon EventBridge. Cela vous permet de lancer l'exécution de votre pipeline de création de modèle en fonction de n'importe quel événement dans votre bus d'événements. Vous pouvez ainsi automatiser l'exécution de votre pipeline et répondre automatiquement à des événements tels que les modifications du poste de formation ou de l'état des terminaux. EventBridge Les événements incluent le chargement d'un nouveau fichier dans votre compartiment Amazon S3, un changement de statut de votre point de SageMaker terminaison Amazon dû à une dérive et des sujets relatifs au Amazon Simple Notification Service (SNS).

Les actions Pipelines suivantes peuvent être lancées automatiquement : 

Pour plus d'informations sur la planification des SageMaker tâches, consultez Automatiser SageMaker avec Amazon EventBridge.

Planifier un pipeline avec Amazon EventBridge

Pour démarrer l'exécution d'un pipeline avec Amazon CloudWatch Events, vous devez créer une EventBridge règle. Lorsque vous créez une règle pour les événements, vous spécifiez une action cible à entreprendre lorsque EventBridge vous recevez un événement correspondant à la règle. Lorsqu'un événement correspond à la règle, EventBridge envoie l'événement à la cible spécifiée et lance l'action définie dans la règle.

Les didacticiels suivants montrent comment planifier l'exécution d'un pipeline à EventBridge l'aide de la EventBridge console ou du AWS CLI. 

Prérequis

Création d'une EventBridge règle à l'aide de la EventBridge console

La procédure suivante montre comment créer une EventBridge règle à l'aide de la EventBridge console. 

Votre règle est maintenant en vigueur et prête à lancer les exécutions de votre pipeline.

Créez une EventBridge règle à l'aide du AWS CLI

La procédure suivante montre comment créer une EventBridge règle à l'aide du AWS CLI.

Planifier un pipeline avec le SageMaker Python SDK

Les sections suivantes vous montrent comment configurer les autorisations d'accès aux EventBridge ressources et créer votre calendrier de pipeline à l'aide de SageMaker PythonSDK.

Autorisations nécessaires

Vous devez disposer des autorisations nécessaires pour utiliser le planificateur de pipeline. Procédez comme suit pour configurer vos autorisations :

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

Création d'un calendrier de pipeline

À l'aide du PipelineSchedule constructeur, vous pouvez planifier un pipeline pour qu'il s'exécute une seule fois ou à un intervalle prédéterminé. Un calendrier de pipeline doit être du type atrate, oucron. Cet ensemble de types de planification est une extension des options de EventBridge planification. Pour plus d'informations sur l'utilisation de la PipelineSchedule classe, consultez sagemaker.workflow.triggers. PipelineSchedule. L'exemple suivant montre comment créer chaque type de planification avecPipelineSchedule.

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"
)

Attachez le déclencheur à votre pipeline

Pour vous rattacher PipelineSchedule à votre pipeline, put_triggers appelez l'appel sur l'objet de pipeline que vous avez créé avec une liste de déclencheurs. Si vous obtenez une réponseARN, cela signifie que vous avez créé le calendrier avec succès dans votre compte et EventBridge que vous commencez à appeler le pipeline cible à l'heure ou au rythme spécifiés. Vous devez spécifier un rôle doté des autorisations appropriées pour associer des déclencheurs à un pipeline parent. Si vous n'en fournissez pas, Pipelines extrait le rôle par défaut utilisé pour créer le pipeline à partir du fichier de configuration.

L'exemple suivant montre comment associer un calendrier à un pipeline.

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>)

Décrire les déclencheurs actuels

Pour récupérer des informations sur les déclencheurs de pipeline que vous avez créés, vous pouvez les invoquer describe_trigger() API avec le nom du déclencheur. Cette commande renvoie des détails sur l'expression de planification créée, tels que son heure de début, son état activé et d'autres informations utiles. L'extrait suivant montre un exemple d'invocation :

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

Nettoyer les ressources du déclencheur

Avant de supprimer votre pipeline, nettoyez les déclencheurs existants pour éviter une fuite de ressources dans votre compte. Vous devez supprimer les déclencheurs avant de détruire le pipeline parent. Vous pouvez supprimer vos déclencheurs en transmettant une liste de noms de déclencheurs au delete_triggersAPI. L'extrait suivant montre comment supprimer des déclencheurs.

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