coverage filename sometime changes when from cache

brownben · Nov 8, 2024 · 6d84eb3 · 6d84eb3
1 parent f90f312
commit 6d84eb3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/monitoring.py b/src/monitoring.py
@@ -22,7 +22,9 @@ def should_trace_file(filename):
 
     def line_callback(self, code_object, line_number):
         if id(code_object) in self.trace:
-            self.lines[code_object.co_filename].add(line_number)
+            lines = self.lines.get(code_object.co_filename)
+            if lines:
+                lines.add(line_number)
             return sys.monitoring.DISABLE
 
         if Tracer.should_trace_file(code_object.co_filename):
@@ -37,6 +39,7 @@ def line_callback(self, code_object, line_number):
     def get_lines(self):
         return dict(self.lines)
 
+
 tracer = Tracer()
 
 sys.monitoring.use_tool_id(sys.monitoring.COVERAGE_ID, "xc")