Skip to content

Commit 2f75244

Browse files
committed
fix: add arg docstring
1 parent 3f29553 commit 2f75244

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

src/sagemaker/hyperparameters.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ def retrieve_default(
4646
retrieve the default hyperparameters. (Default: None).
4747
model_version (str): The version of the model for which to retrieve the
4848
default hyperparameters. (Default: None).
49+
instance_type (str): An instance type to optionally supply in order to get hyperparameters
50+
specific for the instance type.
4951
include_container_hyperparameters (bool): ``True`` if the container hyperparameters
5052
should be returned. Container hyperparameters are not used to tune
5153
the specific algorithm. They are used by SageMaker Training jobs to set up

src/sagemaker/jumpstart/artifacts/hyperparameters.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,8 @@ def _retrieve_default_hyperparameters(
6464
object, used for SageMaker interactions. If not
6565
specified, one is created using the default AWS configuration
6666
chain. (Default: sagemaker.jumpstart.constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION).
67+
instance_type (str): An instance type to optionally supply in order to get hyperparameters
68+
specific for the instance type.
6769
Returns:
6870
dict: the hyperparameters to use for the model.
6971
"""

0 commit comments

Comments
 (0)