code formatting

Author: brunopistone

src/sagemaker/remote_function/job.py

@@ -1001,7 +1001,12 @@ def _get_job_name(job_settings, func):
 
 
 def _prepare_and_upload_runtime_scripts(
-    spark_config: SparkConfig, s3_base_uri: str, s3_kms_key: str, sagemaker_session: Session, use_torchrun: bool = False, nproc_per_node: int = 1
+        spark_config: SparkConfig,
+        s3_base_uri: str,
+        s3_kms_key: str,
+        sagemaker_session: Session,
+        use_torchrun: bool = False,
+        nproc_per_node: int = 1,
 ):
     """Copy runtime scripts to a folder and upload to S3.
 

tests/integ/sagemaker/remote_function/test_decorator.py

@@ -819,15 +819,21 @@ def test_decorator_auto_capture(sagemaker_session, auto_capture_test_container):
     )
     subprocess.check_output(shlex.split(cmd), stderr=subprocess.STDOUT).decode("utf-8")
 
-def test_decorator_torchrun(sagemaker_session, dummy_container_without_error, gpu_instance_type, use_torchrun=True, nproc_per_node=2):
+def test_decorator_torchrun(
+        sagemaker_session,
+        dummy_container_without_error,
+        gpu_instance_type,
+        use_torchrun=True,
+        nproc_per_node=2,
+):
     @remote(
         role=ROLE,
         image_uri=dummy_container_without_error,
         instance_type=gpu_instance_type,
         sagemaker_session=sagemaker_session,
         keep_alive_period_in_seconds=60,
         use_torchrun=use_torchrun,
-        nproc_per_node=nproc_per_node
+        nproc_per_node=nproc_per_node,
     )
     def divide(x, y):
         return x / y