fix hive proxy cursor method

monte-carlo-data · Nov 4, 2024 · 9d1b40d · 9d1b40d
1 parent f510c4f
commit 9d1b40d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apollo/integrations/db/hive_proxy_client.py b/apollo/integrations/db/hive_proxy_client.py
@@ -22,7 +22,7 @@ def cursor(self):
         # or the cursor is closed. GetLog() is not automatically called so spark databricks works.
         # With False the cursor will not have a rowcount for DML statements, this is fine for MC.
         # https://github.com/cloudera/impyla/blob/e4c76169f7e5765c09b11c92fceb862dbb9b72be/impala/hiveserver2.py#L122
-        return super().cursor(self, close_finished_queries=False)
+        return self._connection.cursor(close_finished_queries=False)
 
 
 class HiveProxyClient(BaseDbProxyClient):