Configurer une règle d'échec

Le paramètre failure_policy de la classe aiplatform.PipelineJob vous permet de configurer si un pipeline doit échouer immédiatement si l'une des tâches échoue.

Les configurations de règles d'échec suivantes sont compatibles avec Vertex AI Pipelines :

  • Échouer rapidement (failure_policy = 'fast') : choisissez cette configuration si vous souhaitez que le pipeline arrête de planifier de nouvelles tâches après l'échec d'une tâche. Les tâches déjà planifiées continuent de s'exécuter jusqu'à ce qu'elles soient terminées.

  • Échouer lentement (failure_policy = 'slow') : choisissez cette configuration si vous souhaitez que le pipeline continue de s'exécuter même après l'échec d'une tâche. Le pipeline continue de s'exécuter jusqu'à ce que toutes les tâches soient exécutées.

Utilisez l'exemple de code suivant pour configurer la règle d'échec d'un pipeline à l'aide du SDK Vertex AI pour Python :


job = aiplatform.PipelineJob(display_name = 'DISPLAY_NAME',
                             template_path = 'COMPILED_PIPELINE_PATH',
                             pipeline_root = 'PIPELINE_ROOT',
                             project = 'PROJECT_ID',
                             location = 'LOCATION',
                             failure_policy = 'FAILURE_POLICY')

Remplacez les éléments suivants :

  • DISPLAY_NAME : nom du pipeline, qui apparaîtra dans la console Google Cloud.

  • COMPILED_PIPELINE_PATH : chemin d'accès au fichier YAML du pipeline compilé. Il peut s'agir d'un chemin d'accès local ou d'un URI Google Cloud Storage.

  • PIPELINE_ROOT : spécifiez un URI Cloud Storage auquel votre compte de service de pipelines peut accéder. Les artefacts des exécutions de votre pipeline sont stockés dans la racine du pipeline.

  • PROJECT_ID: projet Google Cloud dans lequel ce pipeline s'exécute.

  • LOCATION : région dans laquelle le pipeline s'exécute. Pour en savoir plus sur les régions dans lesquelles Vertex AI Pipelines est disponible, consultez le guide des emplacements Vertex AI. Si vous ne définissez pas ce paramètre, l'emplacement par défaut défini dans aiplatform.init est utilisé.

  • FAILURE_POLICY : spécifiez la règle d'échec pour l'ensemble du pipeline. Les configurations suivantes sont disponibles :

    • Pour configurer l'échec du pipeline après l'échec d'une tâche, saisissez fast.

    • Pour configurer le pipeline de sorte qu'il continue à planifier des tâches après l'échec d'une tâche, saisissez slow.

    Si vous ne définissez pas ce paramètre, la configuration de la règle d'échec est définie par défaut sur slow.

Étape suivante