More lint fixes

Author: akuma12

src/sagemaker/processing.py

@@ -1865,9 +1865,7 @@ def _get_codeartifact_command(self, codeartifact_repo_arn: str) -> str:
         )
         m = re.match(arn_regex, codeartifact_repo_arn)
         if not m:
-            raise ValueError(
-                "invalid CodeArtifact repository arn {}".format(codeartifact_repo_arn)
-            )
+            raise ValueError("invalid CodeArtifact repository arn {}".format(codeartifact_repo_arn))
         domain = m.group("domain")
         owner = m.group("account")
         repository = m.group("repository")
@@ -1903,12 +1901,11 @@ def _generate_framework_script(
                 logged into before installing dependencies (default: None).
         """
         if codeartifact_repo_arn:
-            codeartifact_login_command = self._get_codeartifact_command(
-                codeartifact_repo_arn
-            )
+            codeartifact_login_command = self._get_codeartifact_command(codeartifact_repo_arn)
         else:
-            codeartifact_login_command = \
+            codeartifact_login_command = (
                 "echo 'CodeArtifact repository not specified. Skipping login.'"
+            )
 
         return dedent(
             """\
@@ -2031,9 +2028,7 @@ def _create_and_upload_runproc(
         from sagemaker.workflow.utilities import _pipeline_config, hash_object
 
         if _pipeline_config and _pipeline_config.pipeline_name:
-            runproc_file_str = self._generate_framework_script(
-                user_script, codeartifact_repo_arn
-            )
+            runproc_file_str = self._generate_framework_script(user_script, codeartifact_repo_arn)
             runproc_file_hash = hash_object(runproc_file_str)
             s3_uri = s3.s3_path_join(
                 "s3://",

tests/unit/test_processing.py

@@ -1105,7 +1105,9 @@ def test_pyspark_processor_configuration_path_pipeline_config(
 
 @patch("sagemaker.workflow.utilities._pipeline_config", MOCKED_PIPELINE_CONFIG)
 def test_get_codeartifact_command(pipeline_session):
-    codeartifact_repo_arn = "arn:aws:codeartifact:us-west-2:012345678901:repository/test-domain/test-repository"
+    codeartifact_repo_arn = (
+        "arn:aws:codeartifact:us-west-2:012345678901:repository/test-domain/test-repository"
+    )
 
     processor = PyTorchProcessor(
         role=ROLE,
@@ -1116,9 +1118,14 @@ def test_get_codeartifact_command(pipeline_session):
         sagemaker_session=pipeline_session,
     )
 
-    codeartifact_command = processor._get_codeartifact_command(codeartifact_repo_arn=codeartifact_repo_arn)
+    codeartifact_command = processor._get_codeartifact_command(
+        codeartifact_repo_arn=codeartifact_repo_arn
+    )
 
-    assert codeartifact_command == "aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2"  # noqa: E501
+    assert (
+        codeartifact_command
+        == "aws codeartifact login --tool pip --domain test-domain --domain-owner 012345678901 --repository test-repository --region us-west-2"
+    )  # noqa: E501
 
 
 @patch("sagemaker.workflow.utilities._pipeline_config", MOCKED_PIPELINE_CONFIG)
@@ -1193,7 +1200,9 @@ def test_generate_framework_script_with_codeartifact(pipeline_session):
 
     framework_script = processor._generate_framework_script(
         user_script="process.py",
-        codeartifact_repo_arn="arn:aws:codeartifact:us-west-2:012345678901:repository/test-domain/test-repository"
+        codeartifact_repo_arn=(
+            "arn:aws:codeartifact:us-west-2:012345678901:repository/test-domain/test-repository"
+        ),
     )
 
     assert framework_script == dedent(