Update airflow/utils/decorators.py

Co-authored-by: Wei Lee <[email protected]>
apache · Feb 6, 2025 · 6df6641 · 6df6641
1 parent 696e0ea
commit 6df6641
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/airflow/utils/decorators.py b/airflow/utils/decorators.py
@@ -36,7 +36,8 @@ def __init__(self, task_decorator_name: str) -> None:
         }
 
     def _is_task_decorator(self, decorator_node: cst.Decorator) -> bool:
-        if isinstance(decorator_node.decorator, cst.Name):
+        decorator_expr = decorator_node.decorator
+        if isinstance(decorator_expr, cst.Name):
             return decorator_node.decorator.value in self.decorators_to_remove
         elif isinstance(decorator_node.decorator, cst.Attribute) and isinstance(
             decorator_node.decorator.value, cst.Name