Skip to content

Commit 1481489

Browse files
use value on enum (aws#5444)
--- X-AI-Prompt: i have these 3 tests failing that i need help debugging. i'll paste logs/test names in the next prompt. investigate and find the best path forward. from the logs, it's likely because we're referencing the enum instead of its value X-AI-Tool: Cline
1 parent 23a8b36 commit 1481489

1 file changed

Lines changed: 11 additions & 11 deletions

File tree

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)