@@ -277,13 +277,13 @@ def describe_model(
277
277
model_version = get_hub_model_version (
278
278
hub_model_name = model_name ,
279
279
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 ,
281
281
sagemaker_session = self ._sagemaker_session ,
282
282
hub_model_version = model_version ,
283
283
)
284
284
285
285
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 ,
287
287
hub_content_name = model_name ,
288
288
hub_content_version = model_version ,
289
289
hub_content_type = HubContentType .MODEL_REFERENCE .value ,
@@ -297,7 +297,7 @@ def describe_model(
297
297
model_version = get_hub_model_version (
298
298
hub_model_name = model_name ,
299
299
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 ,
301
301
sagemaker_session = self ._sagemaker_session ,
302
302
hub_model_version = model_version ,
303
303
)
0 commit comments