Airflow 2 - ModuleNotFoundError: No module named 'airflow.operators.sensors'

Question

After upgrading to Airflow 2, I got that error in some DAGs:

ModuleNotFoundError: No module named 'airflow.operators.sensors'

user_1029822 · Answer

I resolved by change the import.

from airflow.operators.sensors import BaseSensorOperator

from airflow.sensors import BaseSensorOperator

user_10499624 · Answer

The new one that works:

from airflow.sensors.base import BaseSensorOperator

Chosen answer doesn't work for newer versions of Airflow.

user_1644172 · Answer

For Airflow 2.1.1 I first installed Amazon provider:

pip install apache-airflow-providers-amazon

and then imported S3KeySensor:

from airflow.providers.amazon.aws.sensors.s3_key import S3KeySensor

user_4117496 · Answer

I was trying to import ExternalTaskSensor and my research led me to this post, it turned out to be this class.

The correct import for me was

from airflow.sensors.external_task import ExternalTaskSensor

Just FYI in case anyone runs into this in the future.