I am extracting data from AWS Athena using pyathena library and following function:
def import_ben_datalake(ACCESS_KEY, SECRET_KEY, S3_DIR, REGION, start, end