Merge branch 'main' into kubernetes_on_warning_callback

astronomer · Nov 16, 2023 · 6a153dc · 6a153dc
2 parents 5e79dd8 + 709ba44
commit 6a153dc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/cosmos/config.py b/cosmos/config.py
@@ -258,7 +258,7 @@ class ExecutionConfig:
 
     execution_mode: ExecutionMode = ExecutionMode.LOCAL
     test_indirect_selection: TestIndirectSelection = TestIndirectSelection.EAGER
-    dbt_executable_path: str | Path = get_system_dbt()
+    dbt_executable_path: str | Path = field(default_factory=get_system_dbt)
 
     dbt_project_path: InitVar[str | Path | None] = None
 

diff --git a/docs/getting_started/mwaa.rst b/docs/getting_started/mwaa.rst
@@ -87,8 +87,9 @@ In your ``my_cosmos_dag.py`` file, import the ``DbtDag`` class from Cosmos and c
 
 .. code-block:: python
 
-    from cosmos import DbtDag, ProjectConfig, ProfileConfig
+    from cosmos import DbtDag, ProjectConfig, ProfileConfig, ExecutionConfig
     from cosmos.profiles import PostgresUserPasswordProfileMapping
+    from cosmos.constants import ExecutionMode
 
     profile_config = ProfileConfig(
         profile_name="default",
@@ -99,11 +100,17 @@ In your ``my_cosmos_dag.py`` file, import the ``DbtDag`` class from Cosmos and c
         ),
     )
 
+    execution_config = ExecutionConfig(
+        dbt_executable_path=f"{os.environ['AIRFLOW_HOME']}/dbt_venv/bin/dbt",
+        execution_mode=ExecutionMode.VIRTUALENV,
+    )
+
     my_cosmos_dag = DbtDag(
         project_config=ProjectConfig(
             "<my_dbt_project>",
         ),
         profile_config=profile_config,
+        execution_config=execution_config,
         # normal dag parameters
         schedule_interval="@daily",
         start_date=datetime(2023, 1, 1),

diff --git a/tests/dbt/test_graph.py b/tests/dbt/test_graph.py
@@ -343,7 +343,8 @@ def test_load_via_dbt_ls_without_profile(mock_validate_dbt_command):
     assert err_info.value.args[0] == expected
 
 
-def test_load_via_dbt_ls_with_invalid_dbt_path():
+@patch("cosmos.dbt.executable.shutil.which", return_value=None)
+def test_load_via_dbt_ls_with_invalid_dbt_path(mock_which):
     project_config = ProjectConfig(dbt_project_path=DBT_PROJECTS_ROOT_DIR / DBT_PROJECT_NAME)
     execution_config = ExecutionConfig(dbt_project_path=DBT_PROJECTS_ROOT_DIR / DBT_PROJECT_NAME)
     render_config = RenderConfig(