From 275899009a2c948140f5ab2e8c6769fec0084a10 Mon Sep 17 00:00:00 2001 From: Ziyi Yang <62045232+zyangbi@users.noreply.github.com> Date: Thu, 15 Aug 2024 19:32:54 -0700 Subject: [PATCH] fix: JoinRuns input order in aws step function version of RegressionTuning benchmark (#118) --- .../RegressionTuningJoinRuns/lambda_function.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/benchmarks/aws_step_functions/RegressionTuningStateMachine/LambdaFunctions/RegressionTuningJoinRuns/lambda_function.py b/benchmarks/aws_step_functions/RegressionTuningStateMachine/LambdaFunctions/RegressionTuningJoinRuns/lambda_function.py index c731c44..189fa51 100644 --- a/benchmarks/aws_step_functions/RegressionTuningStateMachine/LambdaFunctions/RegressionTuningJoinRuns/lambda_function.py +++ b/benchmarks/aws_step_functions/RegressionTuningStateMachine/LambdaFunctions/RegressionTuningJoinRuns/lambda_function.py @@ -20,9 +20,16 @@ def lambda_handler(event, context): print(event) n_samples = json.loads(event[0]['body'])['data']['n_samples'] reqID = json.loads(event[0]['body'])['data']['reqID'] - firstModel = json.loads(event[0]['body'])['data']['firstModel'] - secondModel = json.loads(event[1]['body'])['data']['secondModel'] - + firstModel = None + secondModel = None + + for item in event: + data = json.loads(item['body'])['data'] + if 'firstModel' in data: + firstModel = data['firstModel'] + if 'secondModel' in data: + secondModel = data['secondModel'] + modelLists = [] modelLists.append(firstModel) modelLists.append(secondModel)