From 9b6cd9b6a221f5af4402ae73b805cc71fc9a20fc Mon Sep 17 00:00:00 2001 From: Nathan Nowack Date: Thu, 18 Jan 2024 17:10:31 -0600 Subject: [PATCH] run pre-commits --- prefect_aws/credentials.py | 8 ++++---- prefect_aws/s3.py | 2 +- tests/test_credentials.py | 9 ++++----- 3 files changed, 9 insertions(+), 10 deletions(-) diff --git a/prefect_aws/credentials.py b/prefect_aws/credentials.py index 14854e21..7845b74b 100644 --- a/prefect_aws/credentials.py +++ b/prefect_aws/credentials.py @@ -167,9 +167,9 @@ def get_client(self, client_type: Union[str, ClientType], use_cache: bool = Fals if not use_cache: return self.get_boto3_session().client( service_name=client_type, - **self.aws_client_parameters.get_params_override() + **self.aws_client_parameters.get_params_override(), ) - + return _get_client_cached(ctx=self, client_type=client_type) def get_s3_client(self) -> S3Client: @@ -299,9 +299,9 @@ def get_client(self, client_type: Union[str, ClientType], use_cache: bool = Fals if not use_cache: return self.get_boto3_session().client( service_name=client_type, - **self.aws_client_parameters.get_params_override() + **self.aws_client_parameters.get_params_override(), ) - + return _get_client_cached(ctx=self, client_type=client_type) def get_s3_client(self) -> S3Client: diff --git a/prefect_aws/s3.py b/prefect_aws/s3.py index 32e6c044..c9d63113 100644 --- a/prefect_aws/s3.py +++ b/prefect_aws/s3.py @@ -424,7 +424,7 @@ class S3Bucket(WritableFileSystem, WritableDeploymentStorage, ObjectStorageBlock "for reading and writing objects." ), ) - + cache_client: bool = Field( default=False, description=( diff --git a/tests/test_credentials.py b/tests/test_credentials.py index 2a2ce072..2159c6f4 100644 --- a/tests/test_credentials.py +++ b/tests/test_credentials.py @@ -50,9 +50,8 @@ def test_credentials_get_client(credentials, client_type): with mock_s3(): assert isinstance(credentials.get_client(client_type), BaseClient) -@pytest.mark.parametrize( - "client_type", [member.value for member in ClientType] -) + +@pytest.mark.parametrize("client_type", [member.value for member in ClientType]) def test_get_client_cached(client_type): """ Test to ensure that _get_client_cached function returns the same instance @@ -68,7 +67,7 @@ def test_get_client_cached(client_type): assert _get_client_cached.cache_info().hits == 0, "Initial call count should be 0" assert aws_credentials_block.get_client(client_type) is not None - + assert _get_client_cached.cache_info().hits == 0, "Cache should not yet be used" # Call get_client multiple times with the same parameters @@ -78,4 +77,4 @@ def test_get_client_cached(client_type): # Verify that _get_client_cached is called only once due to caching assert _get_client_cached.cache_info().misses == 1 - assert _get_client_cached.cache_info().hits == 2 \ No newline at end of file + assert _get_client_cached.cache_info().hits == 2