[fix] Bad typing for S3ArtifactStorage_clientconfig args (#3276)

aimhubio · Jan 15, 2025 · 3cfb5a6 · 3cfb5a6
1 parent e5e036d
commit 3cfb5a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aim/storage/artifacts/s3_storage.py b/aim/storage/artifacts/s3_storage.py
@@ -73,7 +73,7 @@ def _get_s3_client(self):
         return client
 
 
-def S3ArtifactStorage_factory(**boto3_client_kwargs: dict):
+def S3ArtifactStorage_factory(**boto3_client_kwargs):
     class S3ArtifactStorageCustom(S3ArtifactStorage):
         def _get_s3_client(self):
             import boto3
@@ -88,7 +88,7 @@ def _get_s3_client(self):
     return S3ArtifactStorageCustom
 
 
-def S3ArtifactStorage_clientconfig(**boto3_client_kwargs: dict):
+def S3ArtifactStorage_clientconfig(**boto3_client_kwargs):
     from aim.storage.artifacts import registry
 
     registry.registry['s3'] = S3ArtifactStorage_factory(**boto3_client_kwargs)