fixed docstyle

aws · Jan 15, 2025 · 423c585 · 423c585
1 parent 7d54096
commit 423c585
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/sagemaker/remote_function/client.py b/src/sagemaker/remote_function/client.py
@@ -284,8 +284,9 @@ def remote(
         use_torchrun (bool): Specifies whether to use torchrun for distributed training.
           Defaults to ``False``.
 
-        nproc_per_node (int): Specifies the number of processes per node for distributed training.
-          Defaults to ``1``. This is defined automatically configured on the instance type.
+        nproc_per_node (int): Specifies the number of processes per node for
+          distributed training. Defaults to ``1``.
+          This is defined automatically configured on the instance type.
     """
 
     def _remote(func):
@@ -728,8 +729,9 @@ def __init__(
             use_torchrun (bool): Specifies whether to use torchrun for distributed training.
               Defaults to ``False``.
 
-            nproc_per_node (int): Specifies the number of processes per node for distributed training.
-              Defaults to ``1``. This is defined automatically configured on the instance type.
+            nproc_per_node (int): Specifies the number of processes per node for
+              distributed training. Defaults to ``1``.
+              This is defined automatically configured on the instance type.
         """
         self.max_parallel_jobs = max_parallel_jobs
 

diff --git a/src/sagemaker/remote_function/runtime_environment/bootstrap_runtime_environment.py b/src/sagemaker/remote_function/runtime_environment/bootstrap_runtime_environment.py
@@ -536,13 +536,13 @@ def main(sys_args=None):
 
         if os.path.exists(RESOURCE_CONFIG):
             try:
-                logger.info(f"Found {RESOURCE_CONFIG}")
+                logger.info("Found %s", RESOURCE_CONFIG)
                 with open(RESOURCE_CONFIG, "r") as f:
                     resource_config = json.load(f)
                 set_env(resource_config=resource_config)
             except (json.JSONDecodeError, FileNotFoundError) as e:
                 # Optionally, you might want to log this error
-                logger.info(f"Error processing {RESOURCE_CONFIG}: {str(e)}")
+                logger.info("ERROR: Error processing %s: %s", RESOURCE_CONFIG, str(e))
 
         exit_code = SUCCESS_EXIT_CODE
     except Exception as e:  # pylint: disable=broad-except