Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions docling/utils/profiling.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -47,15 +47,15 @@ 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

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):
Expand Down
Loading