File tree Expand file tree Collapse file tree 3 files changed +2
-4
lines changed
tests/unit/sagemaker/jumpstart Expand file tree Collapse file tree 3 files changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -817,14 +817,12 @@ def deploy(
817817 f"{ EndpointType .INFERENCE_COMPONENT_BASED } is not supported for Proprietary models."
818818 )
819819
820- print (self .additional_model_data_sources )
821820 self .additional_model_data_sources = _add_model_access_configs_to_model_data_sources (
822821 self .additional_model_data_sources ,
823822 deploy_kwargs .model_access_configs ,
824823 deploy_kwargs .model_id ,
825824 deploy_kwargs .region ,
826825 )
827- print (self .additional_model_data_sources )
828826
829827 try :
830828 predictor = super (JumpStartModel , self ).deploy (** deploy_kwargs .to_kwargs_dict ())
Original file line number Diff line number Diff line change @@ -1548,7 +1548,7 @@ def _add_model_access_configs_to_model_data_sources(
15481548 Args:
15491549 model_data_sources (DeploymentConfigMetadata): Model data sources that will be updated
15501550 model_access_configs (DeploymentConfigMetadata): Config holding accept_eula field
1551- model_id (DeploymentConfigMetadata): Jumpstart mode id.
1551+ model_id (DeploymentConfigMetadata): Jumpstart model id.
15521552 region (str): Region where the user is operating in.
15531553 Returns:
15541554 List[Dict[str, Any]]: List of model data sources with accept EULA configs applied
Original file line number Diff line number Diff line change @@ -2253,7 +2253,7 @@ def test_multiple_public_additional_model_data_source_should_pass_through_both(s
22532253 + self .MOCK_PUBLIC_DEPLOY_CONFIG_ADDITIONAL_MODEL_DATA_SOURCE_POST_CALL
22542254 )
22552255
2256- def test_public_additional_model_data_source_with_model_access_config_should_ignored_it (self ):
2256+ def test_public_additional_model_data_source_with_model_access_config_should_ignore_it (self ):
22572257 # WHERE / WHEN
22582258 additional_model_data_sources = utils ._add_model_access_configs_to_model_data_sources (
22592259 model_data_sources = self .MOCK_PUBLIC_DEPLOY_CONFIG_ADDITIONAL_MODEL_DATA_SOURCE_PRE_CALL ,
You can’t perform that action at this time.
0 commit comments