Skip to content

Commit

Permalink
fix airflow kwargs bug (#2195)
Browse files Browse the repository at this point in the history
Signed-off-by: Future-Outlier <[email protected]>
  • Loading branch information
Future-Outlier authored Feb 16, 2024
1 parent d840369 commit d7f312b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion plugins/flytekit-airflow/flytekitplugins/airflow/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit d7f312b

Please sign in to comment.