Airflow 1.10.3中BranchPythonOperator缺ti参数报错解决方法

这个缺参错误是Airflow 1.10.3的版本机制差异导致的：

• Airflow 1.x 系列版本中，PythonOperator、BranchPythonOperator默认不会向python_callable绑定的自定义函数注入任务上下文对象（包含你代码里用到的ti即TaskInstance实例），必须手动开启参数开关才会传参。
• 你定义的_choose_best_model声明了ti入参，但算子没有开启上下文传递，运行时Airflow调用函数时没传这个参数，就触发了缺参报错。
• 上下文自动注入的逻辑在Airflow 2.x版本才改为默认开启，1.10.3版本必须手动配置。

找到choose_best_model算子的定义代码，新增provide_context=True配置项即可，修改后的代码段如下：

choose_best_model = BranchPythonOperator(
    task_id="choose_best_model",
    python_callable=_choose_best_model,
    provide_context=True  # 开启上下文自动传递，运行时会自动把ti等上下文对象传入函数
)

补充说明：如果后续你写的其他Python类型算子的回调函数需要用到ti、execution_date、dag_run这类上下文变量，在Airflow 1.x版本下都必须给对应算子加上provide_context=True配置才能正常获取参数。

你当前代码里的_training_model函数没有用到任何上下文参数，所以三个训练模型的算子不需要修改，只改上述这一处配置，重新触发DAG任务运行即可正常执行分支判断逻辑。

内容的提问来源于stack exchange，提问作者Abdusoli