diff --git a/docling/utils/profiling.py b/docling/utils/profiling.py index 744e04a685..79f86eefce 100644 --- a/docling/utils/profiling.py +++ b/docling/utils/profiling.py @@ -1,5 +1,5 @@ import time -from datetime import datetime +from datetime import datetime, timezone from enum import Enum from typing import TYPE_CHECKING, List @@ -47,7 +47,7 @@ def __init__( scope: ProfilingScope = ProfilingScope.PAGE, ): if settings.debug.profile_pipeline_timings: - if key not in conv_res.timings.keys(): + if key not in conv_res.timings: conv_res.timings[key] = ProfilingItem(scope=scope) self.conv_res = conv_res self.key = key @@ -55,7 +55,9 @@ def __init__( def __enter__(self): if settings.debug.profile_pipeline_timings: self.start = time.monotonic() - self.conv_res.timings[self.key].start_timestamps.append(datetime.utcnow()) + self.conv_res.timings[self.key].start_timestamps.append( + datetime.now(timezone.utc) + ) return self def __exit__(self, *args):