Skip to content

Commit 11e20e5

Browse files
evakravigoelakash
authored andcommitted
chore: improve docstring
1 parent 1c72c74 commit 11e20e5

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

src/sagemaker/jumpstart/artifacts/model_uris.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,8 +106,7 @@ def _retrieve_model_uri(
106106
artifact S3 URI.
107107
model_scope (str): The model type, i.e. what it is used for.
108108
Valid values: "training" and "inference".
109-
instance_type (str): An instance type to optionally supply in order to get
110-
model artifacts specific for the instance type. (Default: None).
109+
instance_type (str): The ML compute instance type for the specified scope. (Default: None).
111110
region (str): Region for which to retrieve model S3 URI. (Default: None).
112111
tolerate_vulnerable_model (bool): True if vulnerable versions of model
113112
specifications should be tolerated (exception not raised). If False, raises an

src/sagemaker/model_uris.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,7 @@ def retrieve(
4545
the model artifact S3 URI.
4646
model_scope (str): The model type.
4747
Valid values: "training" and "inference".
48-
instance_type (str): An instance type to optionally supply in order to get
49-
model artifacts specific for the instance type. (Default: None).
48+
instance_type (str): The ML compute instance type for the specified scope. (Default: None).
5049
tolerate_vulnerable_model (bool): ``True`` if vulnerable versions of model
5150
specifications should be tolerated without raising an exception. If ``False``, raises an
5251
exception if the script used by this version of the model has dependencies with known

0 commit comments

Comments
 (0)