[airflow] branch main updated: Add custom_operator_name attr to `_BranchPythonDecoratedOperator` (#25783)

[po...@apache.org]

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 <48...@users.noreply.github.com>
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,