Merge branch 'main' into reduce-complexity

astronomer · Dec 4, 2023 · deed56e · deed56e
2 parents 9b156b3 + a2c58f5
commit deed56e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/test_example_dags.py b/tests/test_example_dags.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from pathlib import Path
+from functools import cache
 
 import airflow
 import pytest
@@ -37,6 +38,7 @@ def session():
     return get_session()
 
 
+@cache
 def get_dag_bag() -> DagBag:
     """Create a DagBag by adding the files that are not supported to .airflowignore"""
     with open(AIRFLOW_IGNORE_FILE, "w+") as file:

diff --git a/tests/test_example_dags_no_connections.py b/tests/test_example_dags_no_connections.py
@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 from pathlib import Path
+from functools import cache
 
 import airflow
 import pytest
@@ -23,6 +24,7 @@
 }
 
 
+@cache
 def get_dag_bag() -> DagBag:
     """Create a DagBag by adding the files that are not supported to .airflowignore"""
     with open(AIRFLOW_IGNORE_FILE, "w+") as file: