Skip to content

Commit 3f580d8

Browse files
committed
Improve cache() invalidation
1 parent d2635fd commit 3f580d8

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

awswrangler/s3.py

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -24,26 +24,24 @@ def get_fs(session_primitives=None):
2424
aws_access_key_id, aws_secret_access_key, profile_name, config, s3_additional_kwargs = None, None, None, None, None
2525
args = {}
2626

27-
if session_primitives:
28-
if session_primitives.aws_access_key_id:
27+
if session_primitives is not None:
28+
if session_primitives.aws_access_key_id is not None:
2929
aws_access_key_id = session_primitives.aws_access_key_id
30-
if session_primitives.aws_secret_access_key:
30+
if session_primitives.aws_secret_access_key is not None:
3131
aws_secret_access_key = session_primitives.aws_secret_access_key
32-
if session_primitives.profile_name:
32+
if session_primitives.profile_name is not None:
3333
profile_name = session_primitives.profile_name
34-
if session_primitives.botocore_max_retries:
35-
config = {"retries": {"max_attempts": session_primitives.botocore_max_retries}}
36-
if session_primitives.s3_additional_kwargs:
37-
s3_additional_kwargs = session_primitives.s3_additional_kwargs
34+
if session_primitives.botocore_max_retries is not None:
35+
args["config_kwargs"] = {"retries": {"max_attempts": session_primitives.botocore_max_retries}}
36+
if session_primitives.s3_additional_kwargs is not None:
37+
args["s3_additional_kwargs"] = session_primitives.s3_additional_kwargs
3838

3939
if profile_name:
4040
args["profile_name"] = profile_name
4141
elif aws_access_key_id and aws_secret_access_key:
4242
args["key"] = aws_access_key_id,
4343
args["secret"] = aws_secret_access_key
4444

45-
args["config_kwargs"] = config,
46-
args["s3_additional_kwargs"] = s3_additional_kwargs
4745
fs = s3fs.S3FileSystem(**args)
4846
fs.invalidate_cache(path=None)
4947
return fs

0 commit comments

Comments
 (0)