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(
2543
2543
if tags is not None :
2544
2544
edge_packaging_job_request ["Tags" ] = tags
2545
2545
if resource_key is not None :
2546
- edge_packaging_job_request ["ResourceKey" ] = ( resource_key ,)
2546
+ edge_packaging_job_request ["ResourceKey" ] = resource_key
2547
2547
2548
2548
LOGGER .info ("Creating edge-packaging-job with name: %s" , job_name )
2549
2549
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
2851
2851
"OutputConfig"
2852
2852
]["KmsKeyId" ]
2853
2853
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
+ ]
2857
2857
sagemaker_session .sagemaker_client .create_edge_packaging_job .assert_called_with (
2858
2858
RoleArn = expected_role_arn , # provided from config
2859
2859
OutputConfig = {
You can’t perform that action at this time.
0 commit comments