From 8d956969f7241471c6763cfac48e55f3ea1873b0 Mon Sep 17 00:00:00 2001 From: Baraldo Date: Fri, 22 Sep 2023 14:27:28 -0300 Subject: [PATCH] fix error message at test_build_task_groups_with_callbacks --- tests/test_dagbuilder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_dagbuilder.py b/tests/test_dagbuilder.py index 719bd52..1702873 100644 --- a/tests/test_dagbuilder.py +++ b/tests/test_dagbuilder.py @@ -603,7 +603,7 @@ def test_build_task_groups(): def test_build_task_groups_with_callbacks(): td = dagbuilder.DagBuilder("test_dag", DAG_CONFIG_TASK_GROUP_WITH_CALLBACKS, DEFAULT_CONFIG) if version.parse(AIRFLOW_VERSION) < version.parse("2.2.0"): - error_message = "`task_groups` key can only be used with Airflow 2.2.x" + error_message = "`task_groups` with `default_args` key can only be used with Airflow 2.2.x" with pytest.raises(Exception, match=error_message): td.build() else: