github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-13 13:10:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(core): Fix AWEL branch bug (#1640)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-06-18 11:11:43 +08:00
committed by GitHub
parent 49b56b4576
commit ace169ac46
32 changed files with 870 additions and 481 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/docs/awel/awel_tutorial/basic_syntax/2.4_branch_operator.md
View File

@@ -123,7 +123,7 @@ with DAG("awel_branch_operator") as dag:
task = BranchOperator(branches=branch_mapping)
even_task = MapOperator(task_name="even_task", map_function=even_func)
odd_task = MapOperator(task_name="odd_task", map_function=odd_func)
join_task = JoinOperator(combine_function=combine_function)
join_task = JoinOperator(combine_function=combine_function, can_skip_in_branch=False)
input_task >> task >> even_task >> join_task
input_task >> task >> odd_task >> join_task
@@ -133,6 +133,8 @@ print("=" * 80)
print("Second call, input is 6")
assert asyncio.run(join_task.call(call_data=6)) == 60
```
Note: `can_skip_in_branch` is used to control whether current task can be skipped in the branch.
Set it to `False` to prevent the task from being skipped.
And run the following command to execute the code: