eladkal commented on code in PR #60855:
URL: https://github.com/apache/airflow/pull/60855#discussion_r2970241880
##########
providers/celery/src/airflow/providers/celery/executors/celery_executor_utils.py:
##########
@@ -189,12 +189,18 @@ def on_celery_worker_ready(*args, **kwargs):
# and deserialization for us
@app.task(name="execute_workload")
def execute_workload(input: str) -> None:
+ from celery.exceptions import Ignore
from pydantic import TypeAdapter
from airflow.configuration import conf
from airflow.executors import workloads
from airflow.sdk.execution_time.supervisor import supervise
+ try:
+ from airflow.sdk.exceptions import TaskAlreadyRunningError
+ except ImportError:
+ TaskAlreadyRunningError = None # type: ignore[misc,assignment]
+
Review Comment:
@anishgirianish is it possible to keep core changes in this PR and move all
providers/ changes to a new PR?
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
