diff --git a/dbt/adapters/spark/connections.py b/dbt/adapters/spark/connections.py index 41413f355..1848a72a8 100644 --- a/dbt/adapters/spark/connections.py +++ b/dbt/adapters/spark/connections.py @@ -4,7 +4,7 @@ from dbt.adapters.base import Credentials from dbt.adapters.sql import SQLConnectionManager from dbt.contracts.connection import ConnectionState -from dbt.logger import GLOBAL_LOGGER as logger +from dbt.events import AdapterLogger from dbt.utils import DECIMALS from dbt.adapters.spark import __version__ @@ -42,6 +42,8 @@ import base64 import time +logger = AdapterLogger("Spark") + NUMBERS = DECIMALS + (int, float) diff --git a/dbt/adapters/spark/impl.py b/dbt/adapters/spark/impl.py index 6acbcd446..0a8e4c49d 100644 --- a/dbt/adapters/spark/impl.py +++ b/dbt/adapters/spark/impl.py @@ -16,9 +16,11 @@ from dbt.adapters.spark import SparkColumn from dbt.adapters.base import BaseRelation from dbt.clients.agate_helper import DEFAULT_TYPE_TESTER -from dbt.logger import GLOBAL_LOGGER as logger +from dbt.events import AdapterLogger from dbt.utils import executor +logger = AdapterLogger("Spark") + GET_COLUMNS_IN_RELATION_MACRO_NAME = 'get_columns_in_relation' LIST_SCHEMAS_MACRO_NAME = 'list_schemas' LIST_RELATIONS_MACRO_NAME = 'list_relations_without_caching' diff --git a/tests/integration/base.py b/tests/integration/base.py index 50f4ef918..844488595 100644 --- a/tests/integration/base.py +++ b/tests/integration/base.py @@ -23,9 +23,12 @@ from dbt.clients.jinja import template_cache from dbt.config import RuntimeConfig from dbt.context import providers -from dbt.logger import GLOBAL_LOGGER as logger, log_manager +from dbt.logger import log_manager +from dbt.events import AdapterLogger from dbt.contracts.graph.manifest import Manifest +logger = AdapterLogger("Spark") + INITIAL_ROOT = os.getcwd()