preserve lazy import of _colorize

Author: pfackeldey

Lib/profiling/sampling/pstats_collector.py

@@ -1,13 +1,11 @@
 import collections
 import marshal
 import pstats
-from _colorize import get_colors
+lazy from _colorize import get_colors
 
 from .collector import Collector, extract_lineno
 from .constants import MICROSECONDS_PER_SECOND, PROFILING_MODE_CPU
 
-ANSIColors = get_colors()
-
 class PstatsCollector(Collector):
     aggregating = True
 
@@ -179,6 +177,8 @@ def print_stats(self, sort=-1, limit=None, show_summary=True, mode=None):
         }
 
         # Print header with colors and proper alignment
+        ANSIColors = get_colors()
+
         print(f"{ANSIColors.BOLD_BLUE}Profile Stats:{ANSIColors.RESET}")
 
         header_nsamples = f"{ANSIColors.BOLD_BLUE}{'nsamples':>{col_widths['nsamples']}}{ANSIColors.RESET}"
@@ -270,6 +270,8 @@ def _determine_best_unit(max_value):
 
     def _print_summary(self, stats_list, total_samples):
         """Print summary of interesting functions."""
+        ANSIColors = get_colors()
+
         print(
             f"\n{ANSIColors.BOLD_BLUE}Summary of Interesting Functions:{ANSIColors.RESET}"
         )

Lib/profiling/sampling/sample.py

@@ -6,12 +6,10 @@
 import sysconfig
 import time
 from collections import deque
-from _colorize import get_colors
+lazy from _colorize import get_colors
 
 from .binary_collector import BinaryCollector
 
-ANSIColors = get_colors()
-
 @contextlib.contextmanager
 def _pause_threads(unwinder, blocking):
     """Context manager to pause/resume threads around sampling if blocking is True."""
@@ -273,6 +271,8 @@ def _print_realtime_stats(self):
         )  # Max time = Min Hz
 
         # Build cache stats string if stats collection is enabled
+        ANSIColors = get_colors()
+
         cache_stats_str = ""
         if self.collect_stats:
             try:
@@ -306,6 +306,8 @@ def _print_unwinder_stats(self):
         except RuntimeError:
             return  # Stats not enabled
 
+        ANSIColors = get_colors()
+
         print(f"\n{ANSIColors.BOLD_BLUE}{'='*50}{ANSIColors.RESET}")
         print(f"{ANSIColors.BOLD_BLUE}Unwinder Statistics:{ANSIColors.RESET}")
 
@@ -400,6 +402,8 @@ def _print_binary_stats(self, collector):
         except (ValueError, RuntimeError):
             return  # Collector closed or stats unavailable
 
+        ANSIColors = get_colors()
+
         print(f"  {ANSIColors.CYAN}Binary Encoding:{ANSIColors.RESET}")
 
         repeat_records = stats.get('repeat_records', 0)