Skip to content

Commit

Permalink
fix typing
Browse files Browse the repository at this point in the history
  • Loading branch information
ptiurin committed Jan 15, 2025
1 parent fcf327e commit f81ceb3
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions dbt/adapters/firebolt/connections.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
)
from firebolt.client import DEFAULT_API_URL
from firebolt.client.auth import Auth, ClientCredentials, UsernamePassword
from firebolt.common._types import ExtendedType
from firebolt.db import ARRAY, DECIMAL
from firebolt.db import connect as sdk_connect
from firebolt.db.connection import Connection as SDKConnection
Expand Down Expand Up @@ -191,8 +192,8 @@ def cancel(self, connection: Connection) -> None:
raise NotImplementedError('`cancel` is not implemented for this adapter!')

@classmethod
def data_type_code_to_name( # type: ignore[override] # FIR-29423
cls, type_code: Union[type, ARRAY, DECIMAL]
def data_type_code_to_name(
cls, type_code: Union[type, ExtendedType] # type: ignore[override] # FIR-29423
) -> str:
"""
Convert a Firebolt data type code to a string representing the data type.
Expand All @@ -218,6 +219,7 @@ def data_type_code_to_name( # type: ignore[override] # FIR-29423
return 'bytea'
else:
return 'text'
return 'text'


def _determine_auth(credentials: FireboltCredentials) -> Auth:
Expand Down

0 comments on commit f81ceb3

Please sign in to comment.