diff --git a/providers/celery/src/airflow/providers/celery/executors/celery_executor.py b/providers/celery/src/airflow/providers/celery/executors/celery_executor.py index 55888a77d4d98..0d88eb61b8a1d 100644 --- a/providers/celery/src/airflow/providers/celery/executors/celery_executor.py +++ b/providers/celery/src/airflow/providers/celery/executors/celery_executor.py @@ -53,7 +53,7 @@ from airflow.configuration import conf from airflow.exceptions import AirflowProviderDeprecationWarning, AirflowTaskTimeout from airflow.executors.base_executor import BaseExecutor -from airflow.providers.celery.version_compat import AIRFLOW_V_2_8_PLUS, AIRFLOW_V_3_0_PLUS +from airflow.providers.celery.version_compat import AIRFLOW_V_3_0_PLUS from airflow.stats import Stats from airflow.utils.state import TaskInstanceState from celery import states as celery_states @@ -158,10 +158,7 @@ def __getattr__(name): action="store_true", ) -if AIRFLOW_V_2_8_PLUS: - CELERY_CLI_COMMAND_PATH = "airflow.providers.celery.cli.celery_command" -else: - CELERY_CLI_COMMAND_PATH = "airflow.cli.commands.celery_command" +CELERY_CLI_COMMAND_PATH = "airflow.providers.celery.cli.celery_command" CELERY_COMMANDS = ( ActionCommand( diff --git a/providers/celery/src/airflow/providers/celery/version_compat.py b/providers/celery/src/airflow/providers/celery/version_compat.py index bcc1c2f370778..103abb8cde7f5 100644 --- a/providers/celery/src/airflow/providers/celery/version_compat.py +++ b/providers/celery/src/airflow/providers/celery/version_compat.py @@ -26,5 +26,4 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: return airflow_version.major, airflow_version.minor, airflow_version.micro -AIRFLOW_V_2_8_PLUS = get_base_airflow_version_tuple() >= (2, 8, 0) AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)