fix broken unit test

aws · Dec 24, 2024 · e1b9420 · e1b9420
1 parent 2946a48
commit e1b9420
Showing 1 changed file with 2 additions and 2 deletions.
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_reference_from_inputs(hub_arn, model_name, version)
+        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"
     )
 
     version = "*"
     assert (
-        utils.construct_hub_model_arn_reference_from_inputs(hub_arn, model_name, version)
+        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/*"
     )