lib/swa/cli/athena_command.rb in swa-0.7.3 vs lib/swa/cli/athena_command.rb in swa-0.7.4

@@ -116,9 +116,11 @@
 
         private
 
         def wait_for_query(query_execution_id)
           QueryCompletionWaiter.new(client: athena_client).wait(query_execution_id: query_execution_id)
+        rescue Aws::Waiters::Errors::FailureStateError => error
+          signal_error error.response.query_execution.status.state_change_reason
         end
 
       end
 
       subcommand ["workgroups", "wgs"], "Show work-groups" do