Operador de flujo de air Python pasando parámetros.

Estoy tratando de escribir un operador de Python en un DAG de flujo de air y pasar ciertos parámetros al Python que se puede llamar.

Mi código se ve a continuación.

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) 

Pero sigo recibiendo el error de abajo.

TypeError: my_sleeping_function () obtuvo un argumento de palabra clave inesperado ‘dag_run’

No es capaz de averiguar por qué.

Agregue ** kwargs a la lista de parámetros de su operador después de su parámetro de umbral