File tree Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Expand file tree Collapse file tree 1 file changed +12
-3
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments