Skip to content

Commit 9148e70

Browse files
committed
address nits
1 parent f121eb0 commit 9148e70

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff 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())

src/sagemaker/jumpstart/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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

tests/unit/sagemaker/jumpstart/test_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)