Skip to content

Commit fd8a70c

Browse files
committed
updated docstring for sagemaker sdk doc
1 parent 86c9f7d commit fd8a70c

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/sagemaker/remote_function/client.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,8 @@ def remote(
208208
files are accepted and uploaded to S3.
209209
210210
instance_count (int): The number of instances to use. Defaults to 1.
211-
NOTE: Remote function does not support instance_count > 1 for non Spark jobs.
211+
NOTE: Remote function supports instance_count > 1 for Spark jobs, torchrun and
212+
mpirun utilities
212213
213214
instance_type (str): The Amazon Elastic Compute Cloud (EC2) instance type to use to run
214215
the SageMaker job. e.g. ml.c4.xlarge. If not provided, a ValueError is thrown.
@@ -657,7 +658,8 @@ def __init__(
657658
files are accepted and uploaded to S3.
658659
659660
instance_count (int): The number of instances to use. Defaults to 1.
660-
NOTE: Remote function does not support instance_count > 1 for non Spark jobs.
661+
NOTE: Remote function supports instance_count > 1 for Spark jobs, torchrun and
662+
mpirun utilities
661663
662664
instance_type (str): The Amazon Elastic Compute Cloud (EC2) instance type to use to run
663665
the SageMaker job. e.g. ml.c4.xlarge. If not provided, a ValueError is thrown.

0 commit comments

Comments
 (0)