Fix orm query in HiveEngineSpec.handle_cursor (#2699)

Fix #2643
This commit is contained in:
Riccardo Magliocchetti 2017-04-30 19:55:56 +02:00 committed by Maxime Beauchemin
parent 7d88f80a9b
commit a58adc862e

View File

@ -711,7 +711,7 @@ class HiveEngineSpec(PrestoEngineSpec):
)
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