fix: unit tests, clarify_check_step.py with s3_upload_mode value

Author: Satish615

src/sagemaker/image_uri_config/clarify.json

@@ -29,7 +29,7 @@
                     "us-west-2": "306415355426",
                     "us-gov-west-1": "598674086554"
                 },
-                "repository": "cr-clarify-processing"
+                "repository": "sagemaker-clarify-processing"
             }
         }
     }

src/sagemaker/workflow/clarify_check_step.py

@@ -29,6 +29,7 @@
     ModelConfig,
     ModelPredictedLabelConfig,
     SHAPConfig,
+    ProcessingOutputHandler,
     _upload_analysis_config,
     SageMakerClarifyProcessor,
     _set,
@@ -391,7 +392,7 @@ def _generate_processing_job_parameters(
                 source=SageMakerClarifyProcessor._CLARIFY_OUTPUT,
                 destination=data_config.s3_output_path,
                 output_name="analysis_result",
-                s3_upload_mode="EndOfJob",
+                s3_upload_mode=ProcessingOutputHandler.get_s3_upload_mode(analysis_config),
             )
         return dict(config_input=config_input, data_input=data_input, result_output=result_output)
 

tests/unit/test_clarify.py

@@ -1721,11 +1721,11 @@ def test_invalid_analysis_config(data_config, data_bias_config, model_config):
 
 class TestProcessingOutputHandler:
     def test_get_s3_upload_mode_image(self):
-        analysis_config = {"dataset_type": DatasetType.IMAGE}
+        analysis_config = {"dataset_type": DatasetType.IMAGE.value}
         s3_upload_mode = ProcessingOutputHandler.get_s3_upload_mode(analysis_config)
         assert s3_upload_mode == ProcessingOutputHandler.S3UploadMode.CONTINUOUS.value
 
     def test_get_s3_upload_mode_text(self):
-        analysis_config = {"dataset_type": DatasetType.TEXTCSV}
+        analysis_config = {"dataset_type": DatasetType.TEXTCSV.value}
         s3_upload_mode = ProcessingOutputHandler.get_s3_upload_mode(analysis_config)
         assert s3_upload_mode == ProcessingOutputHandler.S3UploadMode.ENDOFJOB.value