fix: MySQL et al. super calls (apache#23971)

(cherry picked from commit 2af76fc)
airbnb · May 10, 2023 · 2f23d53 · 2f23d53
1 parent 7ce8f6d
commit 2f23d53
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/superset/db_engine_specs/mysql.py b/superset/db_engine_specs/mysql.py
@@ -205,9 +205,13 @@ def adjust_engine_params(
         catalog: Optional[str] = None,
         schema: Optional[str] = None,
     ) -> Tuple[URL, Dict[str, Any]]:
-        uri, new_connect_args = super(
-            MySQLEngineSpec, MySQLEngineSpec
-        ).adjust_engine_params(uri, connect_args, catalog, schema)
+        uri, new_connect_args = super().adjust_engine_params(
+            uri,
+            connect_args,
+            catalog,
+            schema,
+        )
+
         if schema:
             uri = uri.set(database=parse.quote(schema, safe=""))
 

diff --git a/superset/db_engine_specs/ocient.py b/superset/db_engine_specs/ocient.py
@@ -261,9 +261,7 @@ def fetch_data(
         cls, cursor: Any, limit: Optional[int] = None
     ) -> List[Tuple[Any, ...]]:
         try:
-            rows: List[Tuple[Any, ...]] = super(OcientEngineSpec, cls).fetch_data(
-                cursor, limit
-            )
+            rows: List[Tuple[Any, ...]] = super().fetch_data(cursor, limit)
         except Exception as exception:
             with OcientEngineSpec.query_id_mapping_lock:
                 del OcientEngineSpec.query_id_mapping[