fix some weird imports

databricks · Nov 19, 2024 · da76ae9 · da76ae9
1 parent f6c9386
commit da76ae9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/unit/test_adapter.py b/tests/unit/test_adapter.py
@@ -8,15 +8,15 @@
 from mock import Mock
 
 import dbt.flags as flags
-from dbt.adapters.databricks import DatabricksAdapter, DatabricksRelation, __version__
+from dbt.adapters.databricks import DatabricksAdapter, __version__
 from dbt.adapters.databricks.column import DatabricksColumn
 from dbt.adapters.databricks.credentials import (
     CATALOG_KEY_IN_SESSION_PROPERTIES,
     DBT_DATABRICKS_HTTP_SESSION_HEADERS,
     DBT_DATABRICKS_INVOCATION_ENV,
 )
 from dbt.adapters.databricks.impl import get_identifier_list_string
-from dbt.adapters.databricks.relation import DatabricksRelationType
+from dbt.adapters.databricks.relation import DatabricksRelation, DatabricksRelationType
 from dbt.adapters.databricks.utils import check_not_found_error
 from dbt.config import RuntimeConfig
 from tests.unit.utils import config_from_parts_or_dicts

diff --git a/tests/unit/test_column.py b/tests/unit/test_column.py
@@ -1,4 +1,4 @@
-from dbt.adapters.databricks import DatabricksColumn
+from dbt.adapters.databricks.column import DatabricksColumn
 
 
 class TestSparkColumn: