Pyathena encryption with kms key is not working

I am querying Athena using PyAthena-1.4.6.

connection = connect(
    s3_staging_dir='s3://' + athena_results_s3_bucket + '/athena_results',
    region_name=region,
    encryption_option='SSE-KMS',
    kms_key=kms_key
)
cursor = connection.cursor()
cursor.execute(sql_query)

I am able to query the data, but the data in s3 staging_dir is not encrypted. I tried all possible values for ‘encryption_opion’. I could not find any documentation for this. `

[chgangaraju]

This following code is working for me

client = boto3.client('athena')
response = client.start_query_execution(
    QueryString=sql_query,
    QueryExecutionContext={
        'Database': database
    },
    ResultConfiguration={
        'OutputLocation': s3_staging_dir,
        'EncryptionConfiguration': {
            'EncryptionOption': 'SSE_KMS',
            'KmsKey': kms_key_arn
        }
    }
)

[chgangaraju]

It is working as expected with PyAthena also.

connection = connect(
    s3_staging_dir='s3://' + athena_results_s3_bucket + '/athena_results',
    region_name=region,
    encryption_option='SSE_KMS',
    kms_key=kms_key
)
cursor = connection.cursor()
cursor.execute(sql_query)

But the results inside s3_staging_dir is not encrypted (With PyAthena and boto3) . But it is actualy encrypted at s3://aws-athena-query-results-<account_id>-<region>/Unsaved/2019/02/28/0a7d28b4-d785-41e7-b0bd-40fc5a87c05e.csv

Is there any way to encrypt this as well?

boto3 issue reference