ModuleNotFoundError: No module named 'airflow.sdk.definitions.notifier' when using a custom notifier

[jnstockley]

Apache Airflow version

3.0.1

If "Other Airflow 2 version" selected, which one?

No response

What happened?

When I create a custom notifier using a similar example that is shown in the docs, I see this error

Traceback (most recent call last):
  File "/home/airflow/.local/lib/python3.12/site-packages/airflow/sdk/__init__.py", line 121, in __getattr__
    mod = importlib.import_module(module_path, __name__)
          ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/importlib/__init__.py", line 90, in import_module
    return _bootstrap._gcd_import(name[level:], package, level)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ModuleNotFoundError: No module named 'airflow.sdk.definitions.notifier'

What you think should happen instead?

There should be no error and the notifier should work when using on_*_callback

How to reproduce

Create a custom notifier and a DAG using any of the on_*_callback parameters

Operating System

Linux

Versions of Apache Airflow Providers

Tried both
apache-airflow-providers-common-compat==1.7.0rc1 and
apache-airflow-providers-common-compat==1.6.1

Deployment

Other Docker-based deployment

Deployment details

Custom Docker image

Anything else?

No response

Are you willing to submit PR?

• Yes I am willing to submit a PR!

Code of Conduct

• I agree to follow this project's Code of Conduct

[eladkal]

I think this is fixed in #50340
@amoghrajesh can you confirm?

[jnstockley]

@eladkal
Thanks for the update, is there a way I can test it to ensure it works in my scenario?

[amoghrajesh]

Hey @jnstockley, yes this has been fixed by: #50340 and is targetted for 3.0.2

If you want to test it, just cherry pick the fix onto your branch and try it out. You do not need the entire fix too, just https://github.com/apache/airflow/pull/50340/files#diff-6a5a5917333afa5e2df0c72cc7d0607aac789d981a11502c10418285d029305eR87 should be enough