Skip to content

Commit 2584dae

Browse files
committed
Fix Athena cancelled query bug.
1 parent 3a4ad83 commit 2584dae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

awswrangler/pandas.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ def read_sql_athena(self, sql, database, s3_output=None):
8585
query_response = self._session.athena.wait_query(
8686
query_execution_id=query_execution_id)
8787
if query_response.get("QueryExecution").get("Status").get(
88-
"State") == "FAILED":
88+
"State") in ["FAILED", "CANCELLED"]:
8989
reason = (query_response.get("QueryExecution").get("Status").get(
9090
"StateChangeReason"))
9191
message_error = f"Query error: {reason}"

0 commit comments

Comments
 (0)