BranchPythonOperator - return task_id with parameters

Airflow
, , ,
1

According to this example, if the task is_accurate has one param ?
How can i mentioned data when return ‘is_accurate’ ?

@task.branch
  def choose_best_model(accuracy):
    data = "Cool"
    if accuracy > 5:
      return 'is_accurate'
    return 'is_inaccurate'

  @task
  def is_accurate(data):
    pass

  @task
  def is_inaccurate():
    pass
2

from what I’ve searched I dont think you can do this directly.

you can however push a parameter to the xcoms, on the choose_best_model and then access it later.

ti.xcom_push(key="branch_data", value="important_data")