airflow_papermill_operator_jupyter_notebook_execution_with_params.py

1import datetime
2
3from airflow import DAG
4from airflow.providers.papermill.operators.papermill import PapermillOperator
5
6with DAG(
7    dag_id='example_papermill_operator',
8    schedule=None,
9    start_date=datetime.datetime(2021, 1, 1),
10    template_searchpath='/usr/local/airflow/include',
11    catchup=False,
12) as dag:
13    # [START howto_operator_papermill]
14    execute_notebook = PapermillOperator(
15        task_id="run_example_notebook",
16        input_nb="/tmp/hello_world.ipynb",
17        output_nb="/tmp/out-{{ execution_date }}.ipynb",
18        parameters={"msgs": "Ran from Airflow at {{ execution_date }}!"},
19    )
20    # [END howto_operator_papermill]