add region name 2

Author: R-Palazzo

sdgym/s3.py

@@ -209,6 +209,7 @@ def _get_s3_client(output_destination, aws_access_key_id=None, aws_secret_access
             's3',
             aws_access_key_id=aws_access_key_id,
             aws_secret_access_key=aws_secret_access_key,
+            region_name=S3_REGION,
         )
     else:
         s3_client = boto3.client('s3')

tests/unit/test_datasets.py

@@ -14,6 +14,7 @@
     get_dataset_paths,
     load_dataset,
 )
+from sdgym.s3 import S3_REGION
 
 
 class AnyConfigWith:
@@ -136,7 +137,10 @@ def test__download_dataset_private_bucket(boto3_mock, tmpdir):
 
     # asserts
     boto3_mock.client.assert_called_once_with(
-        's3', aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key
+        's3',
+        aws_access_key_id=aws_access_key_id,
+        aws_secret_access_key=aws_secret_access_key,
+        region_name=S3_REGION,
     )
     s3_mock.get_object.assert_called_once_with(
         Bucket='my_bucket', Key=f'{modality.upper()}/{dataset}.zip'

tests/unit/test_s3.py

@@ -9,6 +9,7 @@
 from botocore.exceptions import NoCredentialsError
 
 from sdgym.s3 import (
+    S3_REGION,
     _get_s3_client,
     _upload_dataframe_to_s3,
     _upload_pickle_to_s3,
@@ -178,7 +179,10 @@ def test_write_file_s3(boto3_mock):
 
     # asserts
     boto3_mock.client.assert_called_once_with(
-        's3', aws_access_key_id=aws_access_key_id, aws_secret_access_key=aws_secret_access_key
+        's3',
+        aws_access_key_id=aws_access_key_id,
+        aws_secret_access_key=aws_secret_access_key,
+        region_name=S3_REGION,
     )
     s3_mock.put_object.assert_called_once_with(
         Bucket=bucket_name,
@@ -307,6 +311,7 @@ def test__get_s3_client_with_credentials(mock_boto_client):
         's3',
         aws_access_key_id=aws_access_key_id,
         aws_secret_access_key=aws_secret_access_key,
+        region_name=S3_REGION,
     )
     mock_s3_client.head_bucket.assert_called_once_with(Bucket='my-bucket')