File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/unit/sagemaker/jumpstart/hub Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -102,13 +102,13 @@ def test_construct_hub_model_reference_arn_from_inputs():
102
102
103
103
assert (
104
104
utils .construct_hub_model_reference_arn_from_inputs (hub_arn , model_name , version )
105
- == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub /ModelReference/pytorch-ic-imagenet-v2/1.0.2"
105
+ == hub_arn + " /ModelReference/pytorch-ic-imagenet-v2/1.0.2"
106
106
)
107
107
108
108
version = "*"
109
109
assert (
110
110
utils .construct_hub_model_reference_arn_from_inputs (hub_arn , model_name , version )
111
- == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ ModelReference/pytorch-ic-imagenet-v2/*"
111
+ == hub_arn + " ModelReference/pytorch-ic-imagenet-v2/*"
112
112
)
113
113
114
114
You can’t perform that action at this time.
0 commit comments