From 1e3cc316537753555af7f8b9cb255cbe80b7ecbf Mon Sep 17 00:00:00 2001 From: Gerald Manipon Date: Tue, 20 Aug 2024 16:28:04 -0700 Subject: [PATCH] put annotation differently --- src/unity_initiator/cloud/lambda_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/unity_initiator/cloud/lambda_handler.py b/src/unity_initiator/cloud/lambda_handler.py index 1bb75ba..1b32844 100644 --- a/src/unity_initiator/cloud/lambda_handler.py +++ b/src/unity_initiator/cloud/lambda_handler.py @@ -41,8 +41,8 @@ def lambda_handler_base(event, context): f.write(router_cfg) ROUTER = Router(router_file) os.unlink(router_file) - with xray_recorder.capture("execute_actions") as subsegment: - subsegment.put_annotation("payload", event["payload"]) + with xray_recorder.capture("execute_actions"): + xray_recorder.put_annotation("payload", event["payload"]) return ROUTER.execute_actions(event["payload"])