diff --git a/tests/unit/sqlalchemy/test_dialect.py b/tests/unit/sqlalchemy/test_dialect.py index 5ca6e8c6..317482fb 100644 --- a/tests/unit/sqlalchemy/test_dialect.py +++ b/tests/unit/sqlalchemy/test_dialect.py @@ -278,7 +278,7 @@ def test_trino_connection_certificate_auth(): def test_trino_connection_oauth2_auth(): dialect = TrinoDialect() - url = make_url(f'trino://host/?externalAuthentication=true') + url = make_url('trino://host/?externalAuthentication=true') _, cparams = dialect.create_connect_args(url) assert cparams['http_scheme'] == "https" diff --git a/trino/sqlalchemy/dialect.py b/trino/sqlalchemy/dialect.py index b939ee73..46489526 100644 --- a/trino/sqlalchemy/dialect.py +++ b/trino/sqlalchemy/dialect.py @@ -22,7 +22,12 @@ from trino import dbapi as trino_dbapi from trino import logging -from trino.auth import BasicAuthentication, CertificateAuthentication, JWTAuthentication, OAuth2Authentication +from trino.auth import ( + BasicAuthentication, + CertificateAuthentication, + JWTAuthentication, + OAuth2Authentication, +) from trino.dbapi import Cursor from trino.sqlalchemy import compiler, datatype, error