diff --git a/cosmos/dbt/graph.py b/cosmos/dbt/graph.py index 4b33c67b32..fcfff070b8 100644 --- a/cosmos/dbt/graph.py +++ b/cosmos/dbt/graph.py @@ -320,7 +320,7 @@ def load( if self.project.is_manifest_available(): self.load_from_dbt_manifest() else: - if execution_mode == ExecutionMode.LOCAL and self.profile_config: + if self.profile_config and self.project_path: try: self.load_via_dbt_ls() except FileNotFoundError: diff --git a/tests/dbt/test_graph.py b/tests/dbt/test_graph.py index 4591dd03f3..e53d5dd710 100644 --- a/tests/dbt/test_graph.py +++ b/tests/dbt/test_graph.py @@ -369,9 +369,9 @@ def test_load_manifest_with_manifest(mock_load_from_dbt_manifest): "exec_mode,method,expected_function", [ (ExecutionMode.LOCAL, LoadMode.AUTOMATIC, "mock_load_via_dbt_ls"), - (ExecutionMode.VIRTUALENV, LoadMode.AUTOMATIC, "mock_load_via_custom_parser"), - (ExecutionMode.KUBERNETES, LoadMode.AUTOMATIC, "mock_load_via_custom_parser"), - (ExecutionMode.DOCKER, LoadMode.AUTOMATIC, "mock_load_via_custom_parser"), + (ExecutionMode.VIRTUALENV, LoadMode.AUTOMATIC, "mock_load_via_dbt_ls"), + (ExecutionMode.KUBERNETES, LoadMode.AUTOMATIC, "mock_load_via_dbt_ls"), + (ExecutionMode.DOCKER, LoadMode.AUTOMATIC, "mock_load_via_dbt_ls"), (ExecutionMode.LOCAL, LoadMode.DBT_LS, "mock_load_via_dbt_ls"), (ExecutionMode.LOCAL, LoadMode.CUSTOM, "mock_load_via_custom_parser"), ],