python - Airflow Python 运算符传递参数

Question

我正在尝试在 Airflow DAG 中编写 Python 运算符并将某些参数传递给 Python 可调用对象。

我的代码如下所示。

def my_sleeping_function(threshold):
   print(threshold)

fmfdependency = PythonOperator(
   task_id='poke_check',
   python_callable=my_sleeping_function,
   provide_context=True,
   op_kwargs={'threshold': 100},
   dag=dag)

end = BatchEndOperator(
   queue=QUEUE,
   dag=dag)

start.set_downstream(fmfdependency)
fmfdependency.set_downstream(end)

但我不断收到以下错误。

TypeError: my_sleeping_function() got an unexpected keyword argument 'dag_run'

无法弄清楚原因。

Answer 1

在您的阈值参数之后将 **kwargs 添加到您的运算符参数列表中