pass hub_arn to verify_model_region_and_return_specs

Malav Shastri · Malav Shastri · commit d5d8324b6a53 · 2024-08-13T22:49:18.000-04:00
diff --git a/src/sagemaker/jumpstart/factory/model.py b/src/sagemaker/jumpstart/factory/model.py
@@ -259,6 +259,7 @@ def _add_instance_type_to_kwargs(
         sagemaker_session=kwargs.sagemaker_session,
         model_type=kwargs.model_type,
         config_name=kwargs.config_name,
+        hub_arn=kwargs.hub_arn,
     )
 
     if specs.inference_configs and kwargs.config_name not in specs.inference_configs.configs:
@@ -780,6 +781,7 @@ def _add_config_name_to_deploy_kwargs(
             sagemaker_session=temp_session,
             model_type=kwargs.model_type,
             config_name=kwargs.config_name,
+            hub_arn=kwargs.hub_arn,
         )
         default_config_name = _select_inference_config_from_training_config(
             specs=specs, training_config_name=training_config_name

Original file line number	Diff line number	Diff line change
`@@ -259,6 +259,7 @@ def _add_instance_type_to_kwargs(`
`259`	`259`	`sagemaker_session=kwargs.sagemaker_session,`
`260`	`260`	`model_type=kwargs.model_type,`
`261`	`261`	`config_name=kwargs.config_name,`
	`262`	`+ hub_arn=kwargs.hub_arn,`
`262`	`263`	`)`
`263`	`264`
`264`	`265`	`if specs.inference_configs and kwargs.config_name not in specs.inference_configs.configs:`
`@@ -780,6 +781,7 @@ def _add_config_name_to_deploy_kwargs(`
`780`	`781`	`sagemaker_session=temp_session,`
`781`	`782`	`model_type=kwargs.model_type,`
`782`	`783`	`config_name=kwargs.config_name,`
	`784`	`+ hub_arn=kwargs.hub_arn,`
`783`	`785`	`)`
`784`	`786`	`default_config_name = _select_inference_config_from_training_config(`
`785`	`787`	`specs=specs, training_config_name=training_config_name`