fix: datasetype Enum value in clarify processing output

Satish615 · Satish615 · commit ee721533d664 · 2023-03-27T01:28:27.000-07:00
diff --git a/src/sagemaker/clarify.py b/src/sagemaker/clarify.py
@@ -2112,7 +2112,7 @@ def get_s3_upload_mode(cls, analysis_config: Dict[str, Any]) -> str:
         dataset_type = analysis_config["dataset_type"]
         return (
             ProcessingOutputHandler.S3UploadMode.CONTINUOUS.value
-            if dataset_type == DatasetType.IMAGE
+            if dataset_type == DatasetType.IMAGE.value
             else ProcessingOutputHandler.S3UploadMode.ENDOFJOB.value
         )
 

Original file line number	Diff line number	Diff line change
`@@ -2112,7 +2112,7 @@ def get_s3_upload_mode(cls, analysis_config: Dict[str, Any]) -> str:`
`2112`	`2112`	`dataset_type = analysis_config["dataset_type"]`
`2113`	`2113`	`return (`
`2114`	`2114`	`ProcessingOutputHandler.S3UploadMode.CONTINUOUS.value`
`2115`		`- if dataset_type == DatasetType.IMAGE`
	`2115`	`+ if dataset_type == DatasetType.IMAGE.value`
`2116`	`2116`	`else ProcessingOutputHandler.S3UploadMode.ENDOFJOB.value`
`2117`	`2117`	`)`
`2118`	`2118`