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(
817
817
f"{ EndpointType .INFERENCE_COMPONENT_BASED } is not supported for Proprietary models."
818
818
)
819
819
820
- print (self .additional_model_data_sources )
821
820
self .additional_model_data_sources = _add_model_access_configs_to_model_data_sources (
822
821
self .additional_model_data_sources ,
823
822
deploy_kwargs .model_access_configs ,
824
823
deploy_kwargs .model_id ,
825
824
deploy_kwargs .region ,
826
825
)
827
- print (self .additional_model_data_sources )
828
826
829
827
try :
830
828
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(
1548
1548
Args:
1549
1549
model_data_sources (DeploymentConfigMetadata): Model data sources that will be updated
1550
1550
model_access_configs (DeploymentConfigMetadata): Config holding accept_eula field
1551
- model_id (DeploymentConfigMetadata): Jumpstart mode id.
1551
+ model_id (DeploymentConfigMetadata): Jumpstart model id.
1552
1552
region (str): Region where the user is operating in.
1553
1553
Returns:
1554
1554
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
2253
2253
+ self .MOCK_PUBLIC_DEPLOY_CONFIG_ADDITIONAL_MODEL_DATA_SOURCE_POST_CALL
2254
2254
)
2255
2255
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 ):
2257
2257
# WHERE / WHEN
2258
2258
additional_model_data_sources = utils ._add_model_access_configs_to_model_data_sources (
2259
2259
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