update

src/maggma/stores/aws.py

@@ -361,9 +361,8 @@ def _get_session(self):
     def _get_endpoint_url(self):
         if self.ssh_tunnel is None:
             return self.endpoint_url
-        else:
-            host, port = self.ssh_tunnel.local_address
-            return f"http://{host}:{port}"
+        host, port = self.ssh_tunnel.local_address
+        return f"http://{host}:{port}"
 
     def _get_bucket(self):
         """If on the main thread return the bucket created above, else create a new
@@ -385,7 +384,7 @@ def _get_resource_and_bucket(self):
         try:
             resource.meta.client.head_bucket(Bucket=self.bucket)
         except ClientError:
-            raise RuntimeError(f"Bucket not present on AWS")
+            raise RuntimeError("Bucket not present on AWS")
         bucket = resource.Bucket(self.bucket)
 
         return resource, bucket

tests/stores/test_aws.py

@@ -4,11 +4,12 @@
 import boto3
 import pytest
 from botocore.exceptions import ClientError
-from maggma.stores import MemoryStore, MongoStore, S3Store
-from maggma.stores.ssh_tunnel import SSHTunnel
 from moto import mock_s3
 from sshtunnel import BaseSSHTunnelForwarderError
 
+from maggma.stores import MemoryStore, MongoStore, S3Store
+from maggma.stores.ssh_tunnel import SSHTunnel
+
 
 @pytest.fixture()
 def mongostore():