Skip to content

Commit 504c4e1

Browse files
authored
Merge branch 'master-v2' into master-v2-update-telemetry
2 parents d6fdf75 + 1481489 commit 504c4e1

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

tests/integ/sagemaker/jumpstart/constants.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -95,31 +95,31 @@ class ClassLabelFile(str, Enum):
9595
TEST_ASSETS_SPECS: Dict[
9696
Union[InferenceImageFilename, InferenceTabularDataname, ClassLabelFile], str
9797
] = {
98-
InferenceImageFilename.DOG: _to_s3_path(InferenceImageFilename.DOG, _NB_ASSETS_S3_FOLDER),
99-
InferenceImageFilename.CAT: _to_s3_path(InferenceImageFilename.CAT, _NB_ASSETS_S3_FOLDER),
98+
InferenceImageFilename.DOG: _to_s3_path(InferenceImageFilename.DOG.value, _NB_ASSETS_S3_FOLDER),
99+
InferenceImageFilename.CAT: _to_s3_path(InferenceImageFilename.CAT.value, _NB_ASSETS_S3_FOLDER),
100100
InferenceImageFilename.DAISY: _to_s3_path(
101-
InferenceImageFilename.DAISY, f"{_TF_FLOWERS_S3_FOLDER}/daisy"
101+
InferenceImageFilename.DAISY.value, f"{_TF_FLOWERS_S3_FOLDER}/daisy"
102102
),
103103
InferenceImageFilename.DAISY_2: _to_s3_path(
104-
InferenceImageFilename.DAISY_2, f"{_TF_FLOWERS_S3_FOLDER}/daisy"
104+
InferenceImageFilename.DAISY_2.value, f"{_TF_FLOWERS_S3_FOLDER}/daisy"
105105
),
106106
InferenceImageFilename.ROSE: _to_s3_path(
107-
InferenceImageFilename.ROSE, f"{_TF_FLOWERS_S3_FOLDER}/roses"
107+
InferenceImageFilename.ROSE.value, f"{_TF_FLOWERS_S3_FOLDER}/roses"
108108
),
109109
InferenceImageFilename.NAXOS_TAVERNA: _to_s3_path(
110-
InferenceImageFilename.NAXOS_TAVERNA, _NB_ASSETS_S3_FOLDER
110+
InferenceImageFilename.NAXOS_TAVERNA.value, _NB_ASSETS_S3_FOLDER
111111
),
112112
InferenceImageFilename.PEDESTRIAN: _to_s3_path(
113-
InferenceImageFilename.PEDESTRIAN, _NB_ASSETS_S3_FOLDER
113+
InferenceImageFilename.PEDESTRIAN.value, _NB_ASSETS_S3_FOLDER
114114
),
115-
ClassLabelFile.IMAGE_NET: _to_s3_path(ClassLabelFile.IMAGE_NET, _NB_ASSETS_S3_FOLDER),
115+
ClassLabelFile.IMAGE_NET: _to_s3_path(ClassLabelFile.IMAGE_NET.value, _NB_ASSETS_S3_FOLDER),
116116
InferenceTabularDataname.REGRESSION_ONEHOT: _to_s3_path(
117-
InferenceTabularDataname.REGRESSION_ONEHOT, _NB_ASSETS_S3_FOLDER
117+
InferenceTabularDataname.REGRESSION_ONEHOT.value, _NB_ASSETS_S3_FOLDER
118118
),
119119
InferenceTabularDataname.REGRESSION: _to_s3_path(
120-
InferenceTabularDataname.REGRESSION, _NB_ASSETS_S3_FOLDER
120+
InferenceTabularDataname.REGRESSION.value, _NB_ASSETS_S3_FOLDER
121121
),
122122
InferenceTabularDataname.MULTICLASS: _to_s3_path(
123-
InferenceTabularDataname.MULTICLASS, _NB_ASSETS_S3_FOLDER
123+
InferenceTabularDataname.MULTICLASS.value, _NB_ASSETS_S3_FOLDER
124124
),
125125
}

0 commit comments

Comments
 (0)