put annotation differently

unity-sds · Aug 20, 2024 · 1e3cc31 · 1e3cc31
1 parent 94f71d1
commit 1e3cc31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"])