Skip to content

Commit 8f0083b

Browse files
committed
fix naming and messaging
1 parent 277e0b1 commit 8f0083b

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -665,7 +665,7 @@ def deploy(
665665
managed_instance_scaling: Optional[str] = None,
666666
endpoint_type: EndpointType = EndpointType.MODEL_BASED,
667667
routing_config: Optional[Dict[str, Any]] = None,
668-
model_access_configs: Optional[List[ModelAccessConfig]] = None,
668+
model_access_configs: Optional[Dict[str, ModelAccessConfig]] = None,
669669
) -> PredictorBase:
670670
"""Creates endpoint by calling base ``Model`` class `deploy` method.
671671
@@ -766,7 +766,7 @@ def deploy(
766766
ModelAccessConfig, provide a `{ "model_id", ModelAccessConfig(accept_eula=True) }`
767767
to indicate whether model terms of use have been accepted. The `accept_eula` value
768768
must be explicitly defined as `True` in order to accept the end-user license
769-
agreement (EULA) that some. (Default: None)
769+
agreement (EULA) that some models require. (Default: None)
770770
771771
Raises:
772772
MarketplaceModelSubscriptionError: If the caller is not subscribed to the model.

src/sagemaker/jumpstart/utils.py

Lines changed: 5 additions & 5 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_template(
561+
return get_formatted_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_template(model_id: str, region: str, hosting_eula_key: str):
566+
def get_formatted_eula_message_template(model_id: str, region: str, hosting_eula_key: str) -> str:
567567
"""Returns a formatted EULA message."""
568568
return (
569569
f"Model '{model_id}' requires accepting end-user license agreement (EULA). "
@@ -1562,13 +1562,13 @@ def _add_model_access_configs_to_model_data_sources(
15621562
raise ValueError(
15631563
eula_message_template.format(
15641564
model_source="Additional " if model_data_source.get("ChannelName") else "",
1565-
base_eula_message=format_eula_message_template(
1565+
base_eula_message=get_formatted_eula_message_template(
15661566
model_id=model_id, region=region, hosting_eula_key=hosting_eula_key
15671567
),
15681568
model_access_configs_message=(
1569-
" Please add a ModelAccessConfig entry:"
1569+
"Please add a ModelAccessConfig entry:"
15701570
f" {model_access_config_entry} "
1571-
"to model_access_configs to acknowledge the EULA."
1571+
"to model_access_configs to accept the EULA."
15721572
),
15731573
)
15741574
)

0 commit comments

Comments
 (0)