File tree Expand file tree Collapse file tree 1 file changed +2
-0
lines changed
tests/unit/sagemaker/jumpstart/hub Expand file tree Collapse file tree 1 file changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -95,6 +95,7 @@ def test_construct_hub_model_arn_from_inputs():
9595 == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/Model/pytorch-ic-imagenet-v2/*"
9696 )
9797
98+
9899def test_construct_hub_model_reference_arn_from_inputs ():
99100 model_name , version = "pytorch-ic-imagenet-v2" , "1.0.2"
100101 hub_arn = "arn:aws:sagemaker:us-west-2:123456789123:hub/my-mock-hub"
@@ -110,6 +111,7 @@ def test_construct_hub_model_reference_arn_from_inputs():
110111 == "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/*"
111112 )
112113
114+
113115def test_generate_hub_arn_for_init_kwargs ():
114116 hub_name = "my-hub-name"
115117 hub_arn = "arn:aws:sagemaker:us-west-2:12346789123:hub/my-awesome-hub"
You can’t perform that action at this time.
0 commit comments