Skip to content

Commit a508ebf

Browse files
committed
runned linter
1 parent e747737 commit a508ebf

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

src/sagemaker/remote_function/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@
5858

5959
logger = logging_config.get_logger()
6060

61+
6162
@_telemetry_emitter(feature=Feature.REMOTE_FUNCTION, func_name="remote_function.remote")
6263
def remote(
6364
_func=None,

tests/integ/sagemaker/remote_function/test_decorator.py

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

822+
822823
def test_decorator_torchrun(
823824
sagemaker_session,
824825
dummy_container_without_error,
825826
gpu_instance_type,
826-
use_torchrun=True,
827-
nproc_per_node=2,
827+
use_torchrun=False,
828+
nproc_per_node=1,
828829
):
829830
@remote(
830831
role=ROLE,

0 commit comments

Comments
 (0)