Skip to content

Commit bf706ad

Browse files
committed
fix doc strings
1 parent 4d1e12b commit bf706ad

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

src/sagemaker/jumpstart/utils.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1542,10 +1542,19 @@ def _add_model_access_configs_to_model_data_sources(
15421542
model_access_configs: Dict[str, ModelAccessConfig],
15431543
model_id: str,
15441544
region: str,
1545-
):
1546-
"""Sets AcceptEula to True for gated speculative decoding models"""
1547-
print(model_data_sources)
1545+
) -> List[Dict[str, any]]:
1546+
"""Iterate over the accept EULA configs to ensure all channels are matched
15481547
1548+
Args:
1549+
model_data_sources (DeploymentConfigMetadata): Model data sources that will be updated
1550+
model_access_configs (DeploymentConfigMetadata): Config holding accept_eula field
1551+
model_id (DeploymentConfigMetadata): Jumpstart mode id.
1552+
region (str): Region where the user is operating in.
1553+
Returns:
1554+
List[Dict[str, Any]]: List of model data sources with accept EULA configs applied
1555+
Raise:
1556+
ValueError if at least one channel that requires EULA acceptance as not passed.
1557+
"""
15491558
if not model_data_sources:
15501559
return model_data_sources
15511560

0 commit comments

Comments
 (0)