fixup! fixup! fixup! fixup! gh-138709: Implement CPU time profiling in profiling.sample

Author: pablogsal

Lib/profiling/sampling/sample.py

@@ -611,7 +611,7 @@ def sample(
             collector = CollapsedStackCollector(skip_idle=skip_idle)
             filename = filename or f"collapsed.{pid}.txt"
         case "flamegraph":
-            collector = FlamegraphCollector()
+            collector = FlamegraphCollector(skip_idle=skip_idle)
             filename = filename or f"flamegraph.{pid}.html"
         case _:
             raise ValueError(f"Invalid output format: {output_format}")

Lib/profiling/sampling/stack_collector.py

@@ -50,8 +50,8 @@ def export(self, filename):
 
 
 class FlamegraphCollector(StackTraceCollector):
-    def __init__(self):
-        super().__init__()
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
         self.stats = {}
 
     def set_stats(self, sample_interval_usec, duration_sec, sample_rate, error_rate=None):