Linting

Author: martinRenou

src/sagemaker/estimator.py

@@ -585,7 +585,9 @@ def __init__(
             or security_group_ids is not None
             and subnets is None
         ):
-            raise RuntimeError("When setting up custom VPC, both subnets and security_group_ids must be set")
+            raise RuntimeError(
+                "When setting up custom VPC, both subnets and security_group_ids must be set"
+            )
 
         if self.instance_type in ("local", "local_gpu"):
             if self.instance_type == "local_gpu" and self.instance_count > 1:

tests/unit/test_estimator.py

@@ -517,7 +517,7 @@ def test_subnets_without_security_groups(sagemaker_session):
         DummyFramework(
             entry_point=SCRIPT_PATH,
             sagemaker_session=sagemaker_session,
-            subnets=['123'],
+            subnets=["123"],
         )
 
 
@@ -526,7 +526,7 @@ def test_security_groups_without_subnets(sagemaker_session):
         DummyFramework(
             entry_point=SCRIPT_PATH,
             sagemaker_session=sagemaker_session,
-            security_group_ids=['123'],
+            security_group_ids=["123"],
         )