fix broken unit test

kc1998dp · kc1998dp · commit e8dfd574aeea · 2024-12-23T12:18:21.000-05:00
diff --git a/tests/unit/sagemaker/jumpstart/hub/test_utils.py b/tests/unit/sagemaker/jumpstart/hub/test_utils.py
@@ -100,13 +100,13 @@ def test_construct_hub_model_reference_arn_from_inputs():
     hub_arn = "arn:aws:sagemaker:us-west-2:123456789123:hub/my-mock-hub"
 
     assert (
-        utils.construct_hub_model_arn_from_inputs(hub_arn, model_name, version)
+        utils.construct_hub_model_arn_reference_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"
     )
 
     version = "*"
     assert (
-        utils.construct_hub_model_arn_from_inputs(hub_arn, model_name, version)
+        utils.construct_hub_model_arn_reference_from_inputs(hub_arn, model_name, version)
         == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/*"
     )
 

Original file line number	Diff line number	Diff line change
`@@ -100,13 +100,13 @@ def test_construct_hub_model_reference_arn_from_inputs():`
`100`	`100`	`hub_arn = "arn:aws:sagemaker:us-west-2:123456789123:hub/my-mock-hub"`
`101`	`101`
`102`	`102`	`assert (`
`103`		`- utils.construct_hub_model_arn_from_inputs(hub_arn, model_name, version)`
	`103`	`+ utils.construct_hub_model_arn_reference_from_inputs(hub_arn, model_name, version)`
`104`	`104`	`== "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/1.0.2"`
`105`	`105`	`)`
`106`	`106`
`107`	`107`	`version = "*"`
`108`	`108`	`assert (`
`109`		`- utils.construct_hub_model_arn_from_inputs(hub_arn, model_name, version)`
	`109`	`+ utils.construct_hub_model_arn_reference_from_inputs(hub_arn, model_name, version)`
`110`	`110`	`== "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/*"`
`111`	`111`	`)`
`112`	`112`