From eefc3893111d22b4454a8ba47c9bc1434a2fea5c Mon Sep 17 00:00:00 2001 From: Tomek Urbaszek Date: Tue, 21 Jul 2020 11:36:23 +0200 Subject: [PATCH] Fix dag.clear usages after change from #9824 --- TESTING.rst | 3 ++- docs/executor/debug.rst | 3 ++- tests/test_utils/system_tests_class.py | 3 ++- 3 files changed, 6 insertions(+), 3 deletions(-) diff --git a/TESTING.rst b/TESTING.rst index 6ff4994128bda..fad736a6c27f7 100644 --- a/TESTING.rst +++ b/TESTING.rst @@ -979,7 +979,8 @@ It will run a backfill job: .. code-block:: python if __name__ == '__main__': - dag.clear(reset_dag_runs=True) + from airflow.utils.state import State + dag.clear(dag_run_state=State.NONE) dag.run() diff --git a/docs/executor/debug.rst b/docs/executor/debug.rst index 93218749480ab..c029826fe1004 100644 --- a/docs/executor/debug.rst +++ b/docs/executor/debug.rst @@ -43,7 +43,8 @@ It will run a backfill job: .. code-block:: python if __name__ == '__main__': - dag.clear(reset_dag_runs=True) + from airflow.utils.state import State + dag.clear(dag_run_state=State.NONE) dag.run() diff --git a/tests/test_utils/system_tests_class.py b/tests/test_utils/system_tests_class.py index f5e8bb6131656..217b2ef2bfa64 100644 --- a/tests/test_utils/system_tests_class.py +++ b/tests/test_utils/system_tests_class.py @@ -26,6 +26,7 @@ from airflow.models.dagbag import DagBag from airflow.utils.file import mkdirs from airflow.utils.log.logging_mixin import LoggingMixin +from airflow.utils.state import State from tests.test_utils import AIRFLOW_MAIN_FOLDER from tests.utils.logging_command_executor import get_executor @@ -149,7 +150,7 @@ def run_dag(self, dag_id: str, dag_folder: str = DEFAULT_DAG_FOLDER) -> None: ) self.log.info("Attempting to run DAG: %s", dag_id) - dag.clear(reset_dag_runs=True) + dag.clear(dag_run_state=State.NONE) try: dag.run(ignore_first_depends_on_past=True, verbose=True) except Exception: