From d7f312bb4aec657f166872488f132fe84b25baf2 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Fri, 16 Feb 2024 19:41:24 +0800 Subject: [PATCH] fix airflow kwargs bug (#2195) Signed-off-by: Future-Outlier --- plugins/flytekit-airflow/flytekitplugins/airflow/task.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins/flytekit-airflow/flytekitplugins/airflow/task.py b/plugins/flytekit-airflow/flytekitplugins/airflow/task.py index 4f009f91c1..17a023dfdb 100644 --- a/plugins/flytekit-airflow/flytekitplugins/airflow/task.py +++ b/plugins/flytekit-airflow/flytekitplugins/airflow/task.py @@ -194,7 +194,7 @@ def _flyte_operator(*args, **kwargs): logging.debug("failed to get the attribute GET_ORIGINAL_TASK from user space params") container_image = kwargs.pop("container_image", None) - task_id = kwargs["task_id"] or cls.__name__ + task_id = kwargs.get("task_id", cls.__name__) config = AirflowObj(module=cls.__module__, name=cls.__name__, parameters=kwargs) if not issubclass(cls, airflow_sensors.BaseSensorOperator) and not _is_deferrable(cls):