Skip to content

Commit f7e1249

Browse files
authored
fix: improve access log in executors (#1469)
improve access log in executors
1 parent a83ed2d commit f7e1249

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/coffea/processor/executor.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from cachetools import LRUCache
3131

3232
from ..nanoevents import NanoEventsFactory, schemas
33+
from ..nanoevents.util import key_to_tuple
3334
from ..util import _exception_chain, _hash, deprecate, rich_bar
3435
from .accumulator import Accumulatable, accumulate, set_accumulator
3536
from .checkpointer import CheckpointerABC
@@ -1532,7 +1533,10 @@ def _work_function(
15321533
if isinstance(file, uproot.ReadOnlyDirectory):
15331534
metrics["bytesread"] = file.file.source.num_requested_bytes
15341535
if schema is not None and issubclass(schema, schemas.BaseSchema):
1535-
metrics["columns"] = set(materialized)
1536+
metrics["columns"] = {
1537+
f"{x.branch}-{key_to_tuple(x.buffer_key)[3]}"
1538+
for x in materialized
1539+
}
15361540
metrics["entries"] = len(events)
15371541
metrics["processtime"] = toc - tic
15381542
out = {"out": out, "metrics": metrics, "processed": {item}}

0 commit comments

Comments
 (0)