Skip to content

Commit 41a2113

Browse files
committed
chore: improve docstring
1 parent f64ca6d commit 41a2113

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

src/sagemaker/jumpstart/model.py

+7-5
Original file line numberDiff line numberDiff line change
@@ -321,21 +321,23 @@ def _create_sagemaker_model(
321321
"""Create a SageMaker Model Entity
322322
323323
Args:
324-
instance_type (str): The EC2 instance type that this Model will be
324+
instance_type (str): Optional. The EC2 instance type that this Model will be
325325
used for, this is only used to determine if the image needs GPU
326-
support or not.
327-
accelerator_type (str): Type of Elastic Inference accelerator to
326+
support or not. (Default: None).
327+
accelerator_type (str): Optional. Type of Elastic Inference accelerator to
328328
attach to an endpoint for model loading and inference, for
329329
example, 'ml.eia1.medium'. If not specified, no Elastic
330-
Inference accelerator will be attached to the endpoint.
330+
Inference accelerator will be attached to the endpoint. (Default: None).
331331
tags (List[dict[str, str]]): Optional. The list of tags to add to
332332
the model. Example: >>> tags = [{'Key': 'tagname', 'Value':
333333
'tagvalue'}] For more information about tags, see
334334
https://boto3.amazonaws.com/v1/documentation
335335
/api/latest/reference/services/sagemaker.html#SageMaker.Client.add_tags
336+
(Default: None).
336337
serverless_inference_config (sagemaker.serverless.ServerlessInferenceConfig):
337-
Specifies configuration related to serverless endpoint. Instance type is
338+
Optional. Specifies configuration related to serverless endpoint. Instance type is
338339
not provided in serverless inference. So this is used to find image URIs.
340+
(Default: None).
339341
kwargs: Keyword arguments coming from the caller. This class does not require
340342
any so they are ignored.
341343
"""

0 commit comments

Comments
 (0)