File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff 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 = {
You can’t perform that action at this time.
0 commit comments