fix: remove unnecessary trace_ctx initialization

jchrostek-dd · claude · jchrostek-dd · commit d5916d6a623c · 2026-03-13T14:18:13.000-04:00
The trace_ctx variable is only used inside the `if should_trace_cold_start:` block, so initializing it to None beforehand is redundant. Also updated comment to explain WHY we capture trace context before finishing spans (to ensure ColdStartTracer gets the correct trace_id). 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
diff --git a/datadog_lambda/wrapper.py b/datadog_lambda/wrapper.py
@@ -312,7 +312,6 @@ def _after(self, event, context):
                 and is_new_sandbox()
                 and not is_managed_instances_mode()
             )
-            trace_ctx = None
             if should_trace_cold_start:
                 trace_ctx = tracer.current_trace_context()
 

Original file line number	Diff line number	Diff line change
`@@ -312,7 +312,6 @@ def _after(self, event, context):`
`312`	`312`	`and is_new_sandbox()`
`313`	`313`	`and not is_managed_instances_mode()`
`314`	`314`	`)`
`315`		`- trace_ctx = None`
`316`	`315`	`if should_trace_cold_start:`
`317`	`316`	`trace_ctx = tracer.current_trace_context()`
`318`	`317`