Skip to content

Commit 7b6aef1

Browse files
committed
renaming of eula fn and error msg
1 parent 694b4f2 commit 7b6aef1

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sagemaker/jumpstart/utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -558,12 +558,12 @@ def get_eula_message(model_specs: JumpStartModelSpecs, region: str) -> str:
558558
"""Returns EULA message to display if one is available, else empty string."""
559559
if model_specs.hosting_eula_key is None:
560560
return ""
561-
return format_eula_message_from_specs(
561+
return format_eula_message_template(
562562
model_id=model_specs.model_id, region=region, hosting_eula_key=model_specs.hosting_eula_key
563563
)
564564

565565

566-
def format_eula_message_from_specs(model_id: str, region: str, hosting_eula_key: str):
566+
def format_eula_message_template(model_id: str, region: str, hosting_eula_key: str):
567567
"""Returns a formatted EULA message."""
568568
return (
569569
f"Model '{model_id}' requires accepting end-user license agreement (EULA). "
@@ -1561,8 +1561,8 @@ def _add_model_access_configs_to_model_data_sources(
15611561
)
15621562
raise ValueError(
15631563
eula_message_template.format(
1564-
model_source="Draft " if model_data_source.get("ChannelName") else "",
1565-
base_eula_message=format_eula_message_from_specs(
1564+
model_source="Additional " if model_data_source.get("ChannelName") else "",
1565+
base_eula_message=format_eula_message_template(
15661566
model_id=model_id, region=region, hosting_eula_key=hosting_eula_key
15671567
),
15681568
model_access_configs_message=(

0 commit comments

Comments
 (0)