From 682c2b44c49dac91ef9f3cc3751107d54ca956b9 Mon Sep 17 00:00:00 2001 From: Christoph Auer Date: Wed, 30 Oct 2024 13:44:24 +0100 Subject: [PATCH] Add start_timestamps to ProfilingItem Signed-off-by: Christoph Auer --- docling/utils/profiling.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docling/utils/profiling.py b/docling/utils/profiling.py index fec47964..0d09f17d 100644 --- a/docling/utils/profiling.py +++ b/docling/utils/profiling.py @@ -1,4 +1,5 @@ import time +from datetime import datetime from enum import Enum from typing import TYPE_CHECKING, List @@ -20,6 +21,7 @@ class ProfilingItem(BaseModel): scope: ProfilingScope count: int = 0 times: List[float] = [] + start_timestamps: List[datetime] = [] def avg(self) -> float: return np.average(self.times) # type: ignore @@ -50,6 +52,7 @@ class TimeRecorder: def __enter__(self): if settings.debug.profile_pipeline_timings: self.start = time.monotonic() + self.conv_res.timings[self.key].start_timestamps.append(datetime.utcnow()) return self def __exit__(self, *args):