diff --git a/airflow-core/tests/unit/models/test_callback.py b/airflow-core/tests/unit/models/test_callback.py index 545361608382d..a9173dbb0c5f6 100644 --- a/airflow-core/tests/unit/models/test_callback.py +++ b/airflow-core/tests/unit/models/test_callback.py @@ -100,7 +100,7 @@ class UnknownCallback: def test_get_metric_info(self): callback = TriggererCallback(TEST_ASYNC_CALLBACK, prefix="deadline_alerts", dag_id=TEST_DAG_ID) callback.data["kwargs"] = {"context": {"dag_id": TEST_DAG_ID}, "email": "test@example.com"} - metric_info = callback.get_metric_info(CallbackState.SUCCESS, "0") + metric_info = callback.get_metric_info(CallbackState.SUCCESS.value, "0") assert metric_info["stat"] == "deadline_alerts.callback_success" assert metric_info["tags"] == {