diff --git a/cosmos/dbt/graph.py b/cosmos/dbt/graph.py index 43092ae67..28362818f 100644 --- a/cosmos/dbt/graph.py +++ b/cosmos/dbt/graph.py @@ -23,9 +23,9 @@ ) from cosmos.dbt.executable import get_system_dbt from cosmos.dbt.parser.project import LegacyDbtProject +from cosmos.dbt.project import create_symlinks from cosmos.dbt.selector import select_nodes from cosmos.log import get_logger -from cosmos.utils import create_symlinks logger = get_logger(__name__) diff --git a/cosmos/utils.py b/cosmos/dbt/project.py similarity index 100% rename from cosmos/utils.py rename to cosmos/dbt/project.py diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py index 94b178303..57379c7ba 100644 --- a/cosmos/operators/local.py +++ b/cosmos/operators/local.py @@ -44,7 +44,7 @@ FullOutputSubprocessResult, ) from cosmos.dbt.parser.output import extract_log_issues, parse_output -from cosmos.utils import create_symlinks +from cosmos.dbt.project import create_symlinks DBT_NO_TESTS_MSG = "Nothing to do" DBT_WARN_MSG = "WARN" diff --git a/tests/test_utils.py b/tests/dbt/test_project.py similarity index 77% rename from tests/test_utils.py rename to tests/dbt/test_project.py index 870f96c72..bd2555c98 100644 --- a/tests/test_utils.py +++ b/tests/dbt/test_project.py @@ -1,7 +1,7 @@ from pathlib import Path -from cosmos.utils import create_symlinks +from cosmos.dbt.project import create_symlinks -DBT_PROJECTS_ROOT_DIR = Path(__file__).parent.parent / "dev/dags/dbt" +DBT_PROJECTS_ROOT_DIR = Path(__file__).parent.parent.parent / "dev/dags/dbt" def test_create_symlinks(tmp_path):