This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 648e224cd4 Add custom_operator_name attr to
`_BranchPythonDecoratedOperator` (#25783)
648e224cd4 is described below
commit 648e224cd455f1e374c58cfa48eb1c0ed69c698d
Author: Josh Fell <[email protected]>
AuthorDate: Thu Aug 18 04:33:22 2022 -0400
Add custom_operator_name attr to `_BranchPythonDecoratedOperator` (#25783)
---
airflow/decorators/branch_python.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/airflow/decorators/branch_python.py
b/airflow/decorators/branch_python.py
index b5876f49c7..aa4a3c5c3d 100644
--- a/airflow/decorators/branch_python.py
+++ b/airflow/decorators/branch_python.py
@@ -45,6 +45,8 @@ class _BranchPythonDecoratedOperator(DecoratedOperator,
BranchPythonOperator):
# there are some cases we can't deepcopy the objects (e.g protobuf).
shallow_copy_attrs: Sequence[str] = ('python_callable',)
+ custom_operator_name: str = "@task.branch"
+
def __init__(
self,
**kwargs,
