Fix hub model reference arn enum bug

kc1998dp · kc1998dp · commit 280efe0142bd · 2024-12-19T10:25:28.000-05:00
diff --git a/src/sagemaker/jumpstart/hub/utils.py b/src/sagemaker/jumpstart/hub/utils.py
@@ -106,7 +106,7 @@ def construct_hub_model_reference_arn_from_inputs(
     info = get_info_from_hub_resource_arn(hub_arn)
     arn = (
         f"arn:{info.partition}:sagemaker:{info.region}:{info.account_id}:hub-content/"
-        f"{info.hub_name}/{HubContentType.MODEL_REFERENCE}/{model_name}/{version}"
+        f"{info.hub_name}/{HubContentType.MODEL_REFERENCE.value}/{model_name}/{version}"
     )
 
     return arn

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def construct_hub_model_reference_arn_from_inputs(`
`106`	`106`	`info = get_info_from_hub_resource_arn(hub_arn)`
`107`	`107`	`arn = (`
`108`	`108`	`f"arn:{info.partition}:sagemaker:{info.region}:{info.account_id}:hub-content/"`
`109`		`- f"{info.hub_name}/{HubContentType.MODEL_REFERENCE}/{model_name}/{version}"`
	`109`	`+ f"{info.hub_name}/{HubContentType.MODEL_REFERENCE.value}/{model_name}/{version}"`
`110`	`110`	`)`
`111`	`111`
`112`	`112`	`return arn`