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():
95
95
== "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/Model/pytorch-ic-imagenet-v2/*"
96
96
)
97
97
98
+
98
99
def test_construct_hub_model_reference_arn_from_inputs ():
99
100
model_name , version = "pytorch-ic-imagenet-v2" , "1.0.2"
100
101
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():
110
111
== "arn:aws:sagemaker:us-west-2:123456789123:hub-content/my-mock-hub/ModelReference/pytorch-ic-imagenet-v2/*"
111
112
)
112
113
114
+
113
115
def test_generate_hub_arn_for_init_kwargs ():
114
116
hub_name = "my-hub-name"
115
117
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