From c55742796a0dad7b17c1fab0da2307ac7539fab9 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 31 Oct 2025 14:57:07 +0700 Subject: [PATCH] fix: cache process per comment --- src/axolotl/telemetry/runtime_metrics.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/axolotl/telemetry/runtime_metrics.py b/src/axolotl/telemetry/runtime_metrics.py index 0b4be85b3..fa83c00a7 100644 --- a/src/axolotl/telemetry/runtime_metrics.py +++ b/src/axolotl/telemetry/runtime_metrics.py @@ -118,6 +118,7 @@ class RuntimeMetricsTracker: """Initialize the runtime metrics tracker.""" self.metrics = RuntimeMetrics(start_time=time.time()) self.telemetry_manager = TelemetryManager.get_instance() + self._process = psutil.Process() def start_epoch(self, epoch: int): """Record the start of a new epoch.""" @@ -181,7 +182,7 @@ class RuntimeMetricsTracker: def update_memory_metrics(self): """Update peak memory usage metrics.""" # CPU memory - cpu_memory = psutil.Process().memory_info().rss + cpu_memory = self._process.memory_info().rss self.metrics.peak_cpu_memory = max(self.metrics.peak_cpu_memory, cpu_memory) # GPU memory (if available) @@ -194,7 +195,7 @@ class RuntimeMetricsTracker: def get_memory_metrics(self) -> dict[str, Any]: """Get the current memory metrics as a dictionary.""" memory_metrics = { - "cpu_memory_bytes": psutil.Process().memory_info().rss, + "cpu_memory_bytes": self._process.memory_info().rss, "peak_cpu_memory_bytes": self.metrics.peak_cpu_memory, }