Worksheet Run Query - 504 Gateway Timeout

[sandeephs1]

Describe the bug

Running query in worksheet errors with the below messages - "Network error occurred", Network error: Failed to fetch"

How to Reproduce

Create a dataset, load dataset with parquet data create a worksheet, run a query to fetch data from the dataset. If the query is limited with 10 or 5000, it gets completed if the limit exceeds certain number like 15000 etc it fails with error "Network error occurred", Network error: Failed to fetch" NOTE: Limit number in the query cannot be determined

Expected behavior

Worksheet should fetch the result and display it

Your project

No response

Screenshots

No response

OS

Mac Sonoma 14.3

Python version

3.9.13

AWS data.all version

2.0.0

Additional context

We noticed, if data is big, Run query ends up with “504 Gateway Timeout” error in Application console, though Athena Completes the query execution.