Skip to content
Prev Previous commit
Next Next commit
Adjust number of concurrent transform
  • Loading branch information
ca-nguyen committed Sep 3, 2021
commit 4ad008f8ee38616e8b020ab2f97ce34e36cc055d
6 changes: 3 additions & 3 deletions tests/integ/test_sagemaker_steps.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,11 +243,11 @@ def test_transform_step_with_placeholder(trained_estimator, sfn_client, sfn_role
'model_name': job_name,
'data': transform_input,
'content_type': "text/csv",
'instance_count': 1,
'instance_type': "ml.m5.large",
'instance_count': INSTANCE_COUNT,
'instance_type': INSTANCE_TYPE,
'split_type': 'Line',
'strategy': 'SingleRecord',
'max_concurrent_transforms': 2,
'max_concurrent_transforms': 1,
'max_payload': 5
}

Expand Down