chore: restore version lookup

Mause · Sep 26, 2023 · 703d2a3 · 703d2a3
1 parent 2d76e6a
commit 703d2a3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/duckdb_engine/__init__.py b/duckdb_engine/__init__.py
@@ -173,8 +173,6 @@ class Dialect(PGDialect_psycopg2):
     supports_comments = False
     supports_sane_rowcount = False
     supports_server_side_cursors = False
-    supports_native_enum = True
-    server_version_info = (8, 0)
     inspector = DuckDBInspector
     # colspecs TODO: remap types to duckdb types
     colspecs = util.update_copy(
@@ -229,7 +227,11 @@ def dbapi(**kwargs: Any) -> Type[DBAPI]:
         return DBAPI
 
     def initialize(self, connection: Any) -> None:
-        pass
+        super().initialize(connection)
+        self.supports_native_enum = True
+
+    def _get_server_version_info(self, connection: "Connection") -> Tuple[int, int]:
+        return (8, 0)
 
     def get_default_isolation_level(self, connection: "Connection") -> None:
         raise NotImplementedError()