diff --git a/tap_athena/client.py b/tap_athena/client.py index c5293c1..4a3f51f 100644 --- a/tap_athena/client.py +++ b/tap_athena/client.py @@ -4,12 +4,8 @@ import typing as t -import sqlalchemy from singer_sdk import SQLConnector, SQLStream -if t.TYPE_CHECKING: - from sqlalchemy.engine import Engine - class AthenaConnector(SQLConnector): """The connector for SQLite. @@ -38,21 +34,6 @@ def get_sqlalchemy_url(self, config: dict[str, t.Any]) -> str: f"schema={config['schema_name']}" ) - def create_engine(self) -> Engine: - """Create a SQLAlchemy engine. - - Returns: - A SQLAlchemy engine. - """ - return sqlalchemy.create_engine( - self.sqlalchemy_url, - echo=False, - # TODO: Enable JSON serialization/deserialization. - # https://github.com/MeltanoLabs/tap-athena/issues/35 - # json_serializer=self.serialize_json, # noqa: ERA001 - # json_deserializer=self.deserialize_json, # noqa: ERA001 - ) - class AthenaStream(SQLStream): """The Stream class for Athena."""