Skip to content

Commit 69b2929

Browse files
committed
fix: pylint
1 parent 8086f6e commit 69b2929

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

src/sagemaker/jumpstart/estimator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -719,7 +719,8 @@ def attach(
719719
if not model_id:
720720
raise ValueError(
721721
f"Cannot infer JumpStart model ID from training job '{training_job_name}'. "
722-
"Please specify JumpStart `model_id` when retrieving Estimator for this training job."
722+
"Please specify JumpStart `model_id` when retrieving Estimator "
723+
"for this training job."
723724
)
724725

725726
model_version = model_version or "*"

src/sagemaker/jumpstart/utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -768,6 +768,10 @@ def _get_jumpstart_model_id_version_from_resource_arn(
768768
resource_arn: str,
769769
sagemaker_session: Session = constants.DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
770770
) -> Tuple[Optional[str], Optional[str]]:
771+
"""Returns the JumpStart model ID and version if in resource tags.
772+
773+
Returns None if model ID or version cannot be inferred from tags.
774+
"""
771775

772776
list_tags_result = sagemaker_session.list_tags(resource_arn)
773777

src/sagemaker/predictor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,8 @@ def retrieve_default(
7777
if not model_id:
7878
raise ValueError(
7979
f"Cannot infer JumpStart model ID from endpoint '{endpoint_name}'. "
80-
"Please specify JumpStart `model_id` when retrieving default predictor for this endpoint."
80+
"Please specify JumpStart `model_id` when retrieving default "
81+
"predictor for this endpoint."
8182
)
8283

8384
model_version = model_version or "*"

0 commit comments

Comments
 (0)