Unable to use pyarrow with sql academy with dynamic credentials

[suuhaasa]

conn_str = "awsathena+arrow://:@athena.{region_name}.amazonaws.com:443/"
"{schema_name}?s3_staging_dir={s3_staging_dir}"
Using this conn_str doesn't pass dynamic credentials to pyarrow s3 filesystem. Getting an error:
sqlalchemy.exc.OperationalError: (pyathena.error.OperationalError) When reading information for key ... AWS Error [code 15]: No response body.
But this works conn_str = "awsathena+rest://:@athena.{region_name}.amazonaws.com:443/"
"{schema_name}?s3_staging_dir={s3_staging_dir}"

[suuhaasa]

used this temporary solution in pyathena/arrow/result_set.py which works

 else:
            b3_session = connection.session._session
            aws_access_key_id = b3_session.get_credentials().access_key
            aws_secret_access_key = b3_session.get_credentials().secret_key
            aws_session_token = b3_session.get_credentials().token
            fs = fs.S3FileSystem(
                access_key=aws_access_key_id,
                secret_key=aws_secret_access_key,
                session_token=aws_session_token,
                region=connection.region_name,
            )

[laughingman7743]

Should I change the implementation to a temporary solution?