fix test

Author: kc1998dp

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

@@ -99,16 +99,17 @@ def test_construct_hub_model_arn_from_inputs():
 def test_construct_hub_model_reference_arn_from_inputs():
     model_name, version = "pytorch-ic-imagenet-v2", "1.0.2"
     hub_arn = "arn:aws:sagemaker:us-west-2:123456789123:hub/my-mock-hub"
-
+    hub_content_arn_prefix = "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub
+    
     assert (
         utils.construct_hub_model_reference_arn_from_inputs(hub_arn, model_name, version)
-        == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/1.0.2"
+        == hub_content_arn_prefix + "/ModelReference/pytorch-ic-imagenet-v2/1.0.2"
     )
 
     version = "*"
     assert (
         utils.construct_hub_model_reference_arn_from_inputs(hub_arn, model_name, version)
-        == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/*"
+        == hub_content_arn_prefix + "/ModelReference/pytorch-ic-imagenet-v2/*"
     )