Skip to content

Commit 82841ea

Browse files
authored
Merge branch 'master' into fix/remove-unnecessary-get-caller-identity-call
2 parents 5c7a9fd + 9caf00f commit 82841ea

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/sagemaker/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2543,7 +2543,7 @@ def package_model_for_edge(
25432543
if tags is not None:
25442544
edge_packaging_job_request["Tags"] = tags
25452545
if resource_key is not None:
2546-
edge_packaging_job_request["ResourceKey"] = (resource_key,)
2546+
edge_packaging_job_request["ResourceKey"] = resource_key
25472547

25482548
LOGGER.info("Creating edge-packaging-job with name: %s", job_name)
25492549
self.sagemaker_client.create_edge_packaging_job(**edge_packaging_job_request)

tests/unit/test_session.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2851,9 +2851,9 @@ def test_create_edge_packaging_with_sagemaker_config_injection(sagemaker_session
28512851
"OutputConfig"
28522852
]["KmsKeyId"]
28532853
expected_tags = SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"]["Tags"]
2854-
expected_resource_key = (
2855-
SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"]["ResourceKey"],
2856-
)
2854+
expected_resource_key = SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"][
2855+
"ResourceKey"
2856+
]
28572857
sagemaker_session.sagemaker_client.create_edge_packaging_job.assert_called_with(
28582858
RoleArn=expected_role_arn, # provided from config
28592859
OutputConfig={

0 commit comments

Comments
 (0)