Skip to content

Commit 65b61a6

Browse files
author
Malav Shastri
committed
fix: describe_model if hub_name has been explicitly provided
1 parent 0f3f434 commit 65b61a6

File tree

1 file changed

+3
-3
lines changed
  • src/sagemaker/jumpstart/hub

1 file changed

+3
-3
lines changed

src/sagemaker/jumpstart/hub/hub.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -277,13 +277,13 @@ def describe_model(
277277
model_version = get_hub_model_version(
278278
hub_model_name=model_name,
279279
hub_model_type=HubContentType.MODEL_REFERENCE.value,
280-
hub_name=self.hub_name,
280+
hub_name=self.hub_name if not hub_name else hub_name,
281281
sagemaker_session=self._sagemaker_session,
282282
hub_model_version=model_version,
283283
)
284284

285285
hub_content_description: Dict[str, Any] = self._sagemaker_session.describe_hub_content(
286-
hub_name=self.hub_name,
286+
hub_name=self.hub_name if not hub_name else hub_name,
287287
hub_content_name=model_name,
288288
hub_content_version=model_version,
289289
hub_content_type=HubContentType.MODEL_REFERENCE.value,
@@ -297,7 +297,7 @@ def describe_model(
297297
model_version = get_hub_model_version(
298298
hub_model_name=model_name,
299299
hub_model_type=HubContentType.MODEL.value,
300-
hub_name=self.hub_name,
300+
hub_name=self.hub_name if not hub_name else hub_name,
301301
sagemaker_session=self._sagemaker_session,
302302
hub_model_version=model_version,
303303
)

0 commit comments

Comments
 (0)