Fix orm query in HiveEngineSpec.handle_cursor (#2699)

Fix #2643
apache · Apr 30, 2017 · a58adc8 · a58adc8
1 parent 7d88f80
commit a58adc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/superset/db_engine_specs.py b/superset/db_engine_specs.py
@@ -711,7 +711,7 @@ def handle_cursor(cls, cursor, query, session):
         )
         polled = cursor.poll()
         while polled.operationState in unfinished_states:
-            query = session.query(type(query)).filter_by(id=query.id)
+            query = session.query(type(query)).filter_by(id=query.id).one()
             if query.status == QueryStatus.STOPPED:
                 cursor.cancel()
                 break