airflow

airflow operator import doesn't seem to work

I am new to airflow and need some direction on this one... I'm creating my first dag that uses a subdag and importing the subdag operator

`from airflow.operators.subdag import SubDagOperator`

however I keep getting the flowing error "Broken DAG: [/usr/local/airflow/dags/POC_Main_DAG.py] No module named 'airflow.operators.subdag'"

I also tried importing the dummy operator ang got the same error. on the other hand the below operators seem to be imported as expected.

from airflow.operators.bash_operator import BashOperator
from airflow.operators.python_operator import PythonOperator
from airflow.operators.mysql_operator import MySqlOperator

appreciate help on resolving this issue thanks in advance!

Solution

  • What version of Airflow are you using?

    If you are using Airflow 1.10.x, use the following:

    from airflow.operators.subdag_operator import SubDagOperator

from airflow.operators.bash_operator import BashOperator
from airflow.operators.python_operator import PythonOperator

    In Airflow >=2.0.0, use the following:

    from airflow.operators.subdag import SubDagOperator

from airflow.operators.bash import BashOperator
from airflow.operators.python import PythonOperator