Skip to content

Commit e747737

Browse files
committed
indentation fixed
1 parent a61d042 commit e747737

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

src/sagemaker/remote_function/job.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1001,12 +1001,12 @@ def _get_job_name(job_settings, func):
10011001

10021002

10031003
def _prepare_and_upload_runtime_scripts(
1004-
spark_config: SparkConfig,
1005-
s3_base_uri: str,
1006-
s3_kms_key: str,
1007-
sagemaker_session: Session,
1008-
use_torchrun: bool = False,
1009-
nproc_per_node: int = 1,
1004+
spark_config: SparkConfig,
1005+
s3_base_uri: str,
1006+
s3_kms_key: str,
1007+
sagemaker_session: Session,
1008+
use_torchrun: bool = False,
1009+
nproc_per_node: int = 1,
10101010
):
10111011
"""Copy runtime scripts to a folder and upload to S3.
10121012

tests/integ/sagemaker/remote_function/test_decorator.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -820,11 +820,11 @@ def test_decorator_auto_capture(sagemaker_session, auto_capture_test_container):
820820
subprocess.check_output(shlex.split(cmd), stderr=subprocess.STDOUT).decode("utf-8")
821821

822822
def test_decorator_torchrun(
823-
sagemaker_session,
824-
dummy_container_without_error,
825-
gpu_instance_type,
826-
use_torchrun=True,
827-
nproc_per_node=2,
823+
sagemaker_session,
824+
dummy_container_without_error,
825+
gpu_instance_type,
826+
use_torchrun=True,
827+
nproc_per_node=2,
828828
):
829829
@remote(
830830
role=ROLE,

0 commit comments

Comments
 (0)