black-format

Author: mollyheamazon

src/sagemaker/modules/configs.py

@@ -88,7 +88,7 @@ class SourceCode(BaseConfig):
 
     Parameters:
         source_dir (Optional[str]):
-            The local directory, s3 uri, or path to tar.gz file stored locally or in s3 that contains 
+            The local directory, s3 uri, or path to tar.gz file stored locally or in s3 that contains
             the source code to be used in the training job container.
         requirements (Optional[str]):
             The path within ``source_dir`` to a ``requirements.txt`` file. If specified, the listed

src/sagemaker/modules/train/model_trainer.py

@@ -413,16 +413,22 @@ def _validate_source_code(self, source_code: Optional[SourceCode]):
                     )
                 if requirements:
                     if not source_dir.endswith(".tar.gz"):
-                        if (not _is_valid_path(f"{source_dir}/{requirements}", path_type="File") 
-                        and not _is_valid_s3_uri(f"{source_dir}/{requirements}", path_type="File")):
+                        if not _is_valid_path(
+                            f"{source_dir}/{requirements}", path_type="File"
+                        ) and not _is_valid_s3_uri(
+                            f"{source_dir}/{requirements}", path_type="File"
+                        ):
                             raise ValueError(
                                 f"Invalid 'requirements': {requirements}. "
                                 + "Must be a valid file within the 'source_dir'.",
                             )
                 if entry_script:
                     if not source_dir.endswith(".tar.gz"):
-                        if (not _is_valid_path(f"{source_dir}/{entry_script}", path_type="File") 
-                        and not _is_valid_s3_uri(f"{source_dir}/{entry_script}", path_type="File")):
+                        if not _is_valid_path(
+                            f"{source_dir}/{entry_script}", path_type="File"
+                        ) and not _is_valid_s3_uri(
+                            f"{source_dir}/{entry_script}", path_type="File"
+                        ):
                             raise ValueError(
                                 f"Invalid 'entry_script': {entry_script}. "
                                 + "Must be a valid file within the 'source_dir'.",
@@ -837,8 +843,8 @@ def _prepare_train_script(
         install_requirements = ""
         if source_code.requirements:
             install_requirements = (
-                "echo 'Installing requirements'\n" +
-                f"$SM_PIP_CMD install -r {source_code.requirements}"
+                "echo 'Installing requirements'\n"
+                + f"$SM_PIP_CMD install -r {source_code.requirements}"
             )
 
         working_dir = ""