Skip to content

Commit

Permalink
ruff fix
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-wang-1990 committed Jan 31, 2025
1 parent 8a8d96c commit 20a199e
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 19 deletions.
3 changes: 1 addition & 2 deletions dbt/adapters/databricks/api_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,7 @@

from dbt.adapters.databricks import utils
from dbt.adapters.databricks.__version__ import version
from dbt.adapters.databricks.credentials import BearerAuth
from dbt.adapters.databricks.credentials import DatabricksCredentials
from dbt.adapters.databricks.credentials import BearerAuth, DatabricksCredentials
from dbt.adapters.databricks.logging import logger

DEFAULT_POLLING_INTERVAL = 10
Expand Down
3 changes: 1 addition & 2 deletions dbt/adapters/databricks/connections.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,8 @@
from dbt.adapters.databricks.__version__ import version as __version__
from dbt.adapters.databricks.api_client import DatabricksApiClient
from dbt.adapters.databricks.credentials import (
BearerAuth,
DatabricksCredentials,
DatabricksCredentialManager,
DatabricksCredentials,
)
from dbt.adapters.databricks.events.connection_events import (
ConnectionAcquire,
Expand Down
23 changes: 8 additions & 15 deletions dbt/adapters/databricks/credentials.py
Original file line number Diff line number Diff line change
@@ -1,27 +1,22 @@
import itertools
import json
import os
import re
import threading
from collections.abc import Iterable
from dataclasses import dataclass
from dataclasses import field
from typing import Any, Optional, Callable, Dict, List, cast
from dataclasses import dataclass, field
from typing import Any, Callable, Dict, List, Optional, cast

from dbt_common.exceptions import DbtConfigError, DbtValidationError
from mashumaro import DataClassDictMixin
from requests import PreparedRequest
from requests.auth import AuthBase

from databricks.sdk import WorkspaceClient
from databricks.sdk.core import Config
from databricks.sdk.core import CredentialsProvider
from databricks.sdk.core import Config, CredentialsProvider
from dbt.adapters.contracts.connection import Credentials
from dbt.adapters.databricks.global_state import GlobalState
from dbt.adapters.databricks.logging import logger

from mashumaro import DataClassDictMixin
from requests import PreparedRequest
from requests.auth import AuthBase
from dbt.adapters.databricks.logging import logger

CATALOG_KEY_IN_SESSION_PROPERTIES = "databricks.catalog"
DBT_DATABRICKS_INVOCATION_ENV_REGEX = re.compile("^[A-z0-9\\-]+$")
EXTRACT_CLUSTER_ID_FROM_HTTP_PATH_REGEX = re.compile(r"/?sql/protocolv1/o/\d+/(.*)")
Expand Down Expand Up @@ -154,10 +149,8 @@ def validate_creds(self) -> None:
self.azure_client_id and not self.azure_client_secret
):
raise DbtConfigError(
(
"The config 'azure_client_id' and 'azure_client_secret' "
"must be both present or both absent"
)
"The config 'azure_client_id' and 'azure_client_secret' "
"must be both present or both absent"
)

@classmethod
Expand Down
1 change: 1 addition & 0 deletions tests/unit/test_idle_config.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from unittest.mock import patch

import pytest
from dbt_common.exceptions import DbtRuntimeError

Expand Down

0 comments on commit 20a199e

Please sign in to comment.