diff --git a/tests/jobs/test_local_task_job.py b/tests/jobs/test_local_task_job.py index a4e7c4f387752..cd2c27556d274 100644 --- a/tests/jobs/test_local_task_job.py +++ b/tests/jobs/test_local_task_job.py @@ -285,6 +285,7 @@ def test_localtaskjob_heartbeat_with_default_impersonation(self, psutil_mock, _, assert ti.pid != job1.task_runner.process.pid job_runner.heartbeat_callback() + @pytest.mark.flaky(reruns=5) @pytest.mark.skip_if_database_isolation_mode # Does not work in db isolation mode def test_heartbeat_failed_fast(self): """ diff --git a/tests/www/views/test_views_tasks.py b/tests/www/views/test_views_tasks.py index 0c947c7553228..0c5b43508e2aa 100644 --- a/tests/www/views/test_views_tasks.py +++ b/tests/www/views/test_views_tasks.py @@ -1052,6 +1052,7 @@ def test_graph_view_doesnt_fail_on_recursion_error(app, dag_maker, admin_client) assert resp.status_code == 200 +@pytest.mark.flaky(reruns=5) def test_get_date_time_num_runs_dag_runs_form_data_graph_view(app, dag_maker, admin_client): """Test the get_date_time_num_runs_dag_runs_form_data function.""" from airflow.www.views import get_date_time_num_runs_dag_runs_form_data