Skip to content

Commit f52351b

Browse files
author
Malav Shastri
committed
fix linting issues
1 parent 7dc1803 commit f52351b

File tree

2 files changed

+11
-6
lines changed

2 files changed

+11
-6
lines changed

src/sagemaker/jumpstart/hub/parser_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ def camel_to_snake(camel_case_string: str) -> str:
2121
"""Converts camelCaseString or UpperCamelCaseString to snake_case_string."""
2222
snake_case_string = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", camel_case_string)
2323
if "-" in snake_case_string:
24-
#remove any hyphen from the string for accurate conversion.
24+
# remove any hyphen from the string for accurate conversion.
2525
snake_case_string = snake_case_string.replace("-", "")
2626
return re.sub("([a-z0-9])([A-Z])", r"\1_\2", snake_case_string).lower()
2727

@@ -43,7 +43,7 @@ def walk_and_apply_json(
4343
def _walk_and_apply_json(json_obj, new):
4444
if isinstance(json_obj, dict) and isinstance(new, dict):
4545
for key, value in json_obj.items():
46-
new_key = apply(key)
46+
new_key = apply(key)
4747
if (stop_keys and new_key not in stop_keys) or stop_keys is None:
4848
if isinstance(value, dict):
4949
new[new_key] = {}

src/sagemaker/jumpstart/utils.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1106,10 +1106,15 @@ def get_jumpstart_configs(
11061106

11071107
if hub_arn:
11081108
return (
1109-
{config_name: metadata_configs.configs[camel_to_snake(snake_to_upper_camel(config_name))] for config_name in config_names}
1110-
if metadata_configs
1111-
else {}
1112-
)
1109+
{
1110+
config_name: metadata_configs.configs[
1111+
camel_to_snake(snake_to_upper_camel(config_name))
1112+
]
1113+
for config_name in config_names
1114+
}
1115+
if metadata_configs
1116+
else {}
1117+
)
11131118
return (
11141119
{config_name: metadata_configs.configs[config_name] for config_name in config_names}
11151120
if metadata_configs

0 commit comments

Comments
 (0)