Address review comments

astronomer · Feb 7, 2025 · 9b88931 · 9b88931
1 parent aacc922
commit 9b88931
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/cosmos/_utils/__init__.py b/cosmos/_utils/__init__.py
diff --git a/cosmos/helpers.py → cosmos/_utils/importer.py b/cosmos/helpers.py → cosmos/_utils/importer.py
diff --git a/cosmos/operators/local.py b/cosmos/operators/local.py
@@ -25,6 +25,7 @@
 from packaging.version import Version
 
 from cosmos import cache, settings
+from cosmos._utils.importer import load_method_from_module
 from cosmos.cache import (
     _copy_cached_package_lockfile_to_project,
     _get_latest_cached_package_lockfile,
@@ -34,7 +35,6 @@
 from cosmos.dataset import get_dataset_alias_name
 from cosmos.dbt.project import get_partial_parse_path, has_non_empty_dependencies_file
 from cosmos.exceptions import AirflowCompatibilityError, CosmosDbtRunError, CosmosValueError
-from cosmos.helpers import load_method_from_module
 from cosmos.settings import remote_target_path, remote_target_path_conn_id
 
 try:

diff --git a/tests/test_helpers.py b/tests/test_helpers.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from cosmos.helpers import load_method_from_module
+from cosmos._utils.importer import load_method_from_module
 
 dummy_module_text = """
 def dummy_method():