From b1394acf450bca4cf9ecc6f942068accfdddb74a Mon Sep 17 00:00:00 2001 From: leake Date: Sun, 2 Jun 2024 08:05:20 -0700 Subject: [PATCH] Fixing exception types to include TypeError, which is what is raised in Python 3.12. Fixes #40011 --- airflow/providers/celery/executors/celery_executor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/airflow/providers/celery/executors/celery_executor.py b/airflow/providers/celery/executors/celery_executor.py index e7819ece49806..0e9ef64b5d1c7 100644 --- a/airflow/providers/celery/executors/celery_executor.py +++ b/airflow/providers/celery/executors/celery_executor.py @@ -349,7 +349,7 @@ def change_state( ) -> None: try: super().change_state(key, state, info, remove_running=remove_running) - except AttributeError: + except (AttributeError, TypeError): # Earlier versions of the BaseExecutor don't accept the remove_running parameter for this method # TODO: Remove when min airflow version >= 2.9.2 super().change_state(key, state, info)