User specified workgroup is not honored in athena.read_sql_query with unload_approach=True

[kopachev]

Describe the bug

workgroup argument of athena.read_sql_query() method is not propagated when the new approach unload_approach=True is used.

As a result code tried to use the default workgroup "primary" which isn't accessible for my IAM role hence call fails.

How to Reproduce

Create new Athena workgroup e.g. "large_scans" in your AWS account and pass it like this:

wr.athena.read_sql_query(
        sql=query,
        workgroup="large_scans",
        ctas_approach=False,
        unload_approach=True,  # Note that all works as expected when unload approach is not used

Put a breakpoint in awswrangler.athena._utils._start_query_execution function and notice value of workgroup argument is None

Expected behavior

The code should use the workgroup supplied in the call instead of the default "primary" one.

Your project

No response

Screenshots

No response

OS

Win

Python version

3.8.10

AWS DataWrangler version

2.14.0

Additional context

No response

[jaidisido]

Thank you for raising this @kopachev. I have raised PR #1190 with a fix. It can be tested from the main branch:

pip uninstall awswrangler -y
pip install git+https:/awslabs/aws-data-wrangler.git@main

until it's available in the next release (2.15.0)