fix: Addresses SA warnings (#97)

firebolt-db · Apr 19, 2024 · 9e1eec9 · 9e1eec9
1 parent e01aa85
commit 9e1eec9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/firebolt_db/firebolt_dialect.py b/src/firebolt_db/firebolt_dialect.py
@@ -119,6 +119,9 @@ class FireboltDialect(default.DefaultDialect):
     supports_empty_insert = False
     supports_unicode_statements = True
     supports_unicode_binds = True
+    supports_statement_cache = False
+    # supports_statement_cache Set to False to disable warning
+    # https://sqlalche.me/e/20/cprf
     returns_unicode_strings = True
     description_encoding = None
     supports_native_boolean = True
@@ -131,7 +134,11 @@ def __init__(
         self.context: Union[ExecutionContext, Dict] = context or {}
 
     @classmethod
-    def dbapi(cls) -> ModuleType:
+    def import_dbapi(cls) -> ModuleType:  # For sqlalchemy >= 2.0.0
+        return dbapi
+
+    @classmethod
+    def dbapi(cls) -> ModuleType:  # Kept for backwards compatibility
         return dbapi
 
     def create_connect_args(self, url: URL) -> Tuple[List, Dict]: