diff --git a/src/sagemaker_core/main/utils.py b/src/sagemaker_core/main/utils.py index 126aa556..8d82d74c 100644 --- a/src/sagemaker_core/main/utils.py +++ b/src/sagemaker_core/main/utils.py @@ -64,16 +64,6 @@ def convert_to_snake_case(entity_name): return re.sub("([a-z0-9])([A-Z])", r"\1_\2", snake_case).lower() -def snake_to_pascal(snake_str): - """ - Convert a snake_case string to PascalCase. - Args: - snake_str (str): The snake_case string to be converted. - Returns: - str: The PascalCase string. - """ - components = snake_str.split("_") - return "".join(x.title() for x in components[0:]) def reformat_file_with_black(filename): @@ -198,6 +188,8 @@ def get_textual_rich_logger(name: str, log_level: str = "INFO") -> logging.Logge SPECIAL_SNAKE_TO_PASCAL_MAPPINGS = { "volume_size_in_g_b": "VolumeSizeInGB", "volume_size_in_gb": "VolumeSizeInGB", + "memory_size_in_mb": "MemorySizeInMB", + "supported_response_mime_types": "SupportedResponseMIMETypes", }