Skip to content

Commit d8b173d

Browse files
author
Malav Shastri
committed
Address comments
1 parent 65b61a6 commit d8b173d

File tree

5 files changed

+6
-6
lines changed

5 files changed

+6
-6
lines changed

src/sagemaker/jumpstart/accessors.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,7 @@ def get_model_specs(
301301

302302
except Exception as ex:
303303
logging.info(
304-
"Recieved exeption while calling APIs for ContentType ModelReference, \
304+
"Received exeption while calling APIs for ContentType ModelReference, \
305305
retrying with ContentType Model: "
306306
+ str(ex)
307307
)

src/sagemaker/jumpstart/hub/hub.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -291,7 +291,7 @@ def describe_model(
291291

292292
except Exception as ex:
293293
logging.info(
294-
"Recieved exeption while calling APIs for ContentType ModelReference, retrying with ContentType Model: "
294+
"Received exeption while calling APIs for ContentType ModelReference, retrying with ContentType Model: "
295295
+ str(ex)
296296
)
297297
model_version = get_hub_model_version(

src/sagemaker/jumpstart/hub/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -196,7 +196,7 @@ def get_hub_model_version(
196196
"""Returns available Jumpstart hub model version
197197
198198
Raises:
199-
ResourceNotFound: If the specified model is not found in the hub.
199+
ClientError: If the specified model is not found in the hub.
200200
"""
201201

202202
try:

src/sagemaker/jumpstart/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -429,10 +429,10 @@ def attach(
429429
cls,
430430
endpoint_name: str,
431431
inference_component_name: Optional[str] = None,
432-
hub_name: Optional[str] = None,
433432
model_id: Optional[str] = None,
434433
model_version: Optional[str] = None,
435434
sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
435+
hub_name: Optional[str] = None,
436436
) -> "JumpStartModel":
437437
"""Attaches a JumpStartModel object to an existing SageMaker Endpoint.
438438

tests/unit/sagemaker/jumpstart/hub/test_hub.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -182,13 +182,13 @@ def test_describe_model_success(mock_describe_hub_content_response, sagemaker_se
182182
hub.describe_model("test-model")
183183

184184
mock_list_hub_content_versions.assert_called_with(
185-
hub_name=HUB_NAME, hub_content_name="test-model", hub_content_type="Model"
185+
hub_name=HUB_NAME, hub_content_name="test-model", hub_content_type="ModelReference"
186186
)
187187
sagemaker_session.describe_hub_content.assert_called_with(
188188
hub_name=HUB_NAME,
189189
hub_content_name="test-model",
190190
hub_content_version="3.0",
191-
hub_content_type="Model",
191+
hub_content_type="ModelReference",
192192
)
193193

194194

0 commit comments

Comments
 (0)