diff --git a/duckdb_engine/__init__.py b/duckdb_engine/__init__.py index 1648de80..7b541b1b 100644 --- a/duckdb_engine/__init__.py +++ b/duckdb_engine/__init__.py @@ -7,8 +7,6 @@ from sqlalchemy.dialects.postgresql.base import PGExecutionContext, PGInspector from sqlalchemy.engine.url import URL -name = "duckdb" # TODO: find if sqlalchemy documents this field anywhere - class DBAPI: paramstyle = "qmark" @@ -196,6 +194,3 @@ def get_view_names( rs = connection.exec_driver_sql(s) return [row[0] for row in rs] - - -dialect = Dialect diff --git a/pyproject.toml b/pyproject.toml index 2bed9e5d..75334258 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -27,7 +27,7 @@ ipython-sql = "^0.4.1" sqlalchemy-stubs = "^0.4" [tool.poetry.plugins."sqlalchemy.dialects"] -duckdb = "duckdb_engine" +duckdb = "duckdb_engine:Dialect" [tool.pytest.ini_options] addopts = "--hypothesis-show-statistics"