Skip to content

Commit d858c63

Browse files
author
Siba Rajendran
committed
Ruff lint
1 parent 3aaf29e commit d858c63

File tree

1 file changed

+6
-12
lines changed

1 file changed

+6
-12
lines changed

src/fmcore/aws/factory/boto_factory.py

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -72,9 +72,7 @@ def __create_session(cls, *, role_arn: str = None, region: str, session_name: st
7272

7373
# Get a botocore session with refreshable credentials
7474
botocore_session = cls.__get_refreshable_session(
75-
role_arn=role_arn,
76-
region=region,
77-
session_name=session_name
75+
role_arn=role_arn, region=region, session_name=session_name
7876
)
7977

8078
return boto3.Session(botocore_session=botocore_session)
@@ -96,16 +94,14 @@ def get_client(cls, *, service_name: str, region: str, role_arn: str = None) ->
9694

9795
if key not in cls.__clients:
9896
session = cls.__create_session(
99-
region=region,
100-
role_arn=role_arn,
101-
session_name=f"{service_name}-Session"
97+
region=region, role_arn=role_arn, session_name=f"{service_name}-Session"
10298
)
10399
cls.__clients[key] = session.client(service_name, region_name=region)
104100

105101
return cls.__clients[key]
106102

107103
@classmethod
108-
def get_async_session(cls, *, service_name: str, region: str, role_arn: str = None) -> aioboto3.Session:
104+
def get_async_session(cls, *, service_name: str, region: str, role_arn: str = None) -> aioboto3.Session:
109105
session_name: str = f"Async-{service_name}-Session"
110106

111107
def refresh():
@@ -119,9 +115,7 @@ def refresh():
119115
}
120116

121117
creds = RefreshableCredentials.create_from_metadata(
122-
metadata=refresh(),
123-
refresh_using=refresh,
124-
method="sts-assume-role"
118+
metadata=refresh(), refresh_using=refresh, method="sts-assume-role"
125119
)
126120

127121
frozen = creds.get_frozen_credentials()
@@ -130,7 +124,7 @@ def refresh():
130124
aws_access_key_id=frozen.access_key,
131125
aws_secret_access_key=frozen.secret_key,
132126
aws_session_token=frozen.token,
133-
region_name=region
127+
region_name=region,
134128
)
135129

136-
return session
130+
return session

0 commit comments

Comments
 (0)