From 9c4d500442d4a62c04dc4aa004ed064af9c8d6a8 Mon Sep 17 00:00:00 2001
From: Joey Zhao <5253430+joeyzhao2018@users.noreply.github.com>
Date: Tue, 14 May 2024 16:26:05 -0400
Subject: [PATCH] safety check before create cold start spans

---
 datadog_lambda/cold_start.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/datadog_lambda/cold_start.py b/datadog_lambda/cold_start.py
index d4d4b8c2..8c21420d 100644
--- a/datadog_lambda/cold_start.py
+++ b/datadog_lambda/cold_start.py
@@ -197,7 +197,8 @@ def trace(self, root_nodes: List[ImportNode] = root_nodes):
 
     def trace_tree(self, import_node: ImportNode, parent_span):
         if (
-            import_node.end_time_ns - import_node.start_time_ns
+            not self.trace_ctx
+            or import_node.end_time_ns - import_node.start_time_ns
             < self.min_duration_ms * 1e6
             or import_node.module_name in self.ignored_libs
         ):