From 3d7970adab4da1015c4c440f38a0c9d101ddbe9d Mon Sep 17 00:00:00 2001 From: Carolyn Nguyen <83104894+ca-nguyen@users.noreply.github.com> Date: Fri, 15 Oct 2021 10:28:00 -0700 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Eliot V Scott Co-authored-by: Shiv Lakshminarayan --- src/stepfunctions/steps/sagemaker.py | 2 +- tests/integ/test_sagemaker_steps.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/stepfunctions/steps/sagemaker.py b/src/stepfunctions/steps/sagemaker.py index e209605..6d8ccc1 100644 --- a/src/stepfunctions/steps/sagemaker.py +++ b/src/stepfunctions/steps/sagemaker.py @@ -276,7 +276,7 @@ def __init__(self, state_id, transformer, job_name, model_name, data, data_type= transform_parameters['Tags'] = tags if isinstance(tags, Placeholder) else tags_dict_to_kv_list(tags) if Field.Parameters.value in kwargs and isinstance(kwargs[Field.Parameters.value], dict): - # Update processing_parameters with input parameters + # Update transform_parameters with input parameters merge_dicts(transform_parameters, kwargs[Field.Parameters.value]) kwargs[Field.Parameters.value] = transform_parameters diff --git a/tests/integ/test_sagemaker_steps.py b/tests/integ/test_sagemaker_steps.py index e3f5724..0230f69 100644 --- a/tests/integ/test_sagemaker_steps.py +++ b/tests/integ/test_sagemaker_steps.py @@ -181,7 +181,7 @@ def test_transform_step(trained_estimator, sfn_client, sfn_role_arn): def test_transform_step_with_placeholder(trained_estimator, sfn_client, sfn_role_arn): - # Create transformer from previously created estimator + # Create transformer from supplied estimator job_name = generate_job_name() pca_transformer = trained_estimator.transformer(instance_count=INSTANCE_COUNT, instance_type=INSTANCE_TYPE)