Skip to content

Commit 3ed37f2

Browse files
committed
fix: pytestdoc failures for s3_upload_mode changes
1 parent 0f7af1a commit 3ed37f2

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

src/sagemaker/clarify.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -2094,9 +2094,7 @@ def _upload_analysis_config(analysis_config_file, s3_output_path, sagemaker_sess
20942094

20952095

20962096
class ProcessingOutputHandler:
2097-
"""Handles the parameters sent in SagemakerProcessor.Processingoutput based on the dataset
2098-
type in analysis_config.
2099-
"""
2097+
"""Class to handle the parameters for SagemakerProcessor.Processingoutput"""
21002098

21012099
class S3UploadMode(Enum):
21022100
"""Enum values for different uplaod modes to s3 bucket"""
@@ -2106,8 +2104,13 @@ class S3UploadMode(Enum):
21062104

21072105
@classmethod
21082106
def get_s3_upload_mode(cls, analysis_config: Dict[str, Any]) -> str:
2109-
"""
2110-
returns the s3_upload mode based on the shap_config values
2107+
"""Fetches s3_upload mode based on the shap_config values
2108+
2109+
Args:
2110+
analysis_config (dict): dict Config following the analysis_config.json format
2111+
2112+
Returns:
2113+
The s3_upload_mode type for the processing output.
21112114
"""
21122115
dataset_type = analysis_config["dataset_type"]
21132116
return (

0 commit comments

Comments
 (0)