[hotfix] sqllab presto

Kevogich · Oct 21, 2016 · e9d4749 · e9d4749
1 parent 93f8e7d
commit e9d4749
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/caravel/db_engine_specs.py b/caravel/db_engine_specs.py
@@ -55,7 +55,7 @@ def convert_dttm(cls, target_type, dttm):
         return "'{}'".format(dttm.strftime('%Y-%m-%d %H:%M:%S'))
 
     @classmethod
-    def handle_cursor(cls, cursor, query):
+    def handle_cursor(cls, cursor, query, session):
         """Handle a live cursor between the execute and fetchall calls
 
         The flow works without this method doing anything, but it allows

diff --git a/caravel/sql_lab.py b/caravel/sql_lab.py
@@ -98,7 +98,7 @@ def handle_error(msg):
     cursor = result_proxy.cursor
     query.status = QueryStatus.RUNNING
     session.flush()
-    db_engine_spec.handle_cursor(cursor, query)
+    db_engine_spec.handle_cursor(cursor, query, session)
 
     cdf = None
     if result_proxy.cursor: