Merge branch 'main' into ga-ubuntu

astronomer · Feb 27, 2025 · d5c9061 · d5c9061
2 parents 52e2427 + 8630cae
commit d5c9061
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/dev/dags/example_cosmos_dbt_build.py b/dev/dags/example_cosmos_dbt_build.py
@@ -6,8 +6,7 @@
 from datetime import datetime
 from pathlib import Path
 
-from cosmos import DbtDag, ProfileConfig, ProjectConfig, RenderConfig
-from cosmos.constants import TestBehavior
+from cosmos import DbtDag, ProfileConfig, ProjectConfig, RenderConfig, TestBehavior
 from cosmos.profiles import PostgresUserPasswordProfileMapping
 
 DEFAULT_DBT_ROOT_PATH = Path(__file__).parent / "dbt"

diff --git a/docs/getting_started/mwaa.rst b/docs/getting_started/mwaa.rst
@@ -91,7 +91,6 @@ In your ``my_cosmos_dag.py`` file, import the ``DbtDag`` class from Cosmos and c
     from datetime import datetime
     from cosmos import DbtDag, ProjectConfig, ProfileConfig, ExecutionConfig
     from cosmos.profiles import PostgresUserPasswordProfileMapping
-    from cosmos.constants import ExecutionMode
 
     profile_config = ProfileConfig(
         profile_name="default",

diff --git a/pyproject.toml b/pyproject.toml
@@ -153,6 +153,7 @@ dependencies = [
     "types-python-dateutil",
     "Werkzeug<3.0.0",
     "dbt-core",
+    "methodtools",
 ]
 pre-install-commands = ["sh scripts/test/pre-install-airflow.sh {matrix:airflow} {matrix:python}"]
 

diff --git a/tests/test_example_dags.py b/tests/test_example_dags.py
@@ -81,6 +81,9 @@ def get_dag_bag() -> DagBag:
         if DBT_VERSION < Version("1.5.0"):
             file.writelines(["example_source_rendering.py\n"])
 
+        # TODO: Fix https://github.com/astronomer/astronomer-cosmos/issues/1568
+        file.writelines("example_cosmos_dbt_build.py\n")
+
     print(".airflowignore contents: ")
     print(AIRFLOW_IGNORE_FILE.read_text())
     db = DagBag(EXAMPLE_DAGS_DIR, include_examples=False)