@@ -95,31 +95,31 @@ class ClassLabelFile(str, Enum):
9595TEST_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