Update baseoperator.py

rahulgoyal2987 · Dec 4, 2024 · 17acc46 · 17acc46
1 parent 5b5f8f4
commit 17acc46
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow/models/baseoperator.py b/airflow/models/baseoperator.py
@@ -364,6 +364,8 @@ def wrapper(self, *args, **kwargs):
             sentinel = kwargs.pop(sentinel_key, None)
 
             if sentinel:
+                if not getattr(cls._sentinel,'callers',None):
+                    cls._sentinel.callers = {}
                 cls._sentinel.callers[sentinel_key] = sentinel
             else:
                 sentinel = cls._sentinel.callers.pop(f"{func.__qualname__.split('.')[0]}__sentinel", None)