Skip to content

Commit ebc5e22

Browse files
committed
Using fstring in test
1 parent 6143783 commit ebc5e22

File tree

2 files changed

+3
-6
lines changed

2 files changed

+3
-6
lines changed

src/stepfunctions/exceptions.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -22,5 +22,4 @@ class MissingRequiredParameter(Exception):
2222

2323

2424
class DuplicateStatesInChain(Exception):
25-
pass
26-
25+
pass

tests/integ/test_sagemaker_steps.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -358,15 +358,15 @@ def test_processing_step(sklearn_processor_fixture, sagemaker_session, sfn_clien
358358
def test_processing_step_with_placeholders(sklearn_processor_fixture, sagemaker_session, sfn_client, sfn_role_arn,
359359
sagemaker_role_arn):
360360
region = boto3.session.Session().region_name
361-
input_data = 's3://sagemaker-sample-data-{}/processing/census/census-income.csv'.format(region)
361+
input_data = f"s3://sagemaker-sample-data-{region}/processing/census/census-income.csv"
362362

363363
input_s3 = sagemaker_session.upload_data(
364364
path=os.path.join(DATA_DIR, 'sklearn_processing'),
365365
bucket=sagemaker_session.default_bucket(),
366366
key_prefix='integ-test-data/sklearn_processing/code'
367367
)
368368

369-
output_s3 = 's3://' + sagemaker_session.default_bucket() + '/integ-test-data/sklearn_processing'
369+
output_s3 = f"s3://{sagemaker_session.default_bucket()}/integ-test-data/sklearn_processing"
370370

371371
inputs = [
372372
ProcessingInput(source=input_data, destination='/opt/ml/processing/input', input_name='input-1'),
@@ -422,7 +422,6 @@ def test_processing_step_with_placeholders(sklearn_processor_fixture, sagemaker_
422422
workflow_graph = Chain([processing_step])
423423

424424
with timeout(minutes=DEFAULT_TIMEOUT_MINUTES):
425-
# Create workflow and check definition
426425
workflow = create_workflow_and_check_definition(
427426
workflow_graph=workflow_graph,
428427
workflow_name=unique_name_from_base("integ-test-processing-step-workflow"),
@@ -449,4 +448,3 @@ def test_processing_step_with_placeholders(sklearn_processor_fixture, sagemaker_
449448

450449
# Cleanup
451450
state_machine_delete_wait(sfn_client, workflow.state_machine_arn)
452-
# End of Cleanup

0 commit comments

Comments
 (0)