airflow.providers.google.cloud.triggers.mlengine

Module Contents

Classes

MLEngineStartTrainingJobTrigger

MLEngineStartTrainingJobTrigger run on the trigger worker to perform starting training job operation.

class airflow.providers.google.cloud.triggers.mlengine.MLEngineStartTrainingJobTrigger(conn_id, job_id, region, poll_interval=4.0, package_uris=None, training_python_module=None, training_args=None, runtime_version=None, python_version=None, job_dir=None, project_id=None, labels=None, gcp_conn_id='google_cloud_default', impersonation_chain=None)[source]

Bases: airflow.triggers.base.BaseTrigger

MLEngineStartTrainingJobTrigger run on the trigger worker to perform starting training job operation.

Parameters
  • conn_id (str) – Reference to google cloud connection id

  • job_id (str) – The ID of the job. It will be suffixed with hash of job configuration

  • project_id (str | None) – Google Cloud Project where the job is running

  • poll_interval (float) – polling period in seconds to check for the status

serialize()[source]

Serializes MLEngineStartTrainingJobTrigger arguments and classpath.

async run()[source]

Gets current job execution status and yields a TriggerEvent.

Was this entry helpful?