SagemakerTrigger
Astronomer Providersfired as deferred class with params to run the task in triggerer.
Access Instructions
Install the Astronomer Providers provider package into your Airflow environment.
Import the module into your DAG file and instantiate it with your desired params.
Parameters
job_namename of the job to check status
job_typeType of the sagemaker job whether it is Transform or Training
response_keyThe key which needs to be look in the response.
poke_intervalpolling period in seconds to check for the status
end_timeTime in seconds to wait for a job run to reach a terminal status.
aws_conn_idAWS connection ID for sagemaker
Documentation
- param job_name
name of the job to check status
- param job_type
Type of the sagemaker job whether it is Transform or Training
- param response_key
The key which needs to be look in the response.
- param poke_interval
polling period in seconds to check for the status
- param end_time
Time in seconds to wait for a job run to reach a terminal status.
- param aws_conn_id
AWS connection ID for sagemaker