Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
eric-wang-1990 committed Jan 31, 2025
1 parent e6a6177 commit 4e3acd0
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 110 deletions.
100 changes: 0 additions & 100 deletions dbt/adapters/databricks/auth.py

This file was deleted.

6 changes: 5 additions & 1 deletion dbt/adapters/databricks/connections.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,11 @@
)
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
from dbt.adapters.databricks.credentials import (
BearerAuth,
DatabricksCredentials,
DatabricksCredentialManager,
)
from dbt.adapters.databricks.events.connection_events import (
ConnectionAcquire,
ConnectionCancel,
Expand Down
16 changes: 7 additions & 9 deletions dbt/adapters/databricks/credentials.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,25 +5,23 @@
import threading
from collections.abc import Iterable
from dataclasses import dataclass
from typing import Any, Optional, Union, cast
from dataclasses import field
from typing import Any, Optional, Callable, Dict, List, cast

import keyring
from dbt_common.exceptions import DbtConfigError, DbtValidationError

from databricks.sdk import WorkspaceClient
from databricks.sdk.core import Config
from databricks.sdk.core import CredentialsProvider
from databricks.sdk.oauth import OAuthClient, SessionCredentials
from dbt.adapters.contracts.connection import Credentials
from dbt.adapters.databricks.auth import m2m_auth, token_auth
from dbt.adapters.databricks.events.credential_events import (
CredentialLoadError,
CredentialSaveError,
CredentialShardEvent,
)
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

0 comments on commit 4e3acd0

Please sign in to comment.