Skip to content

FTAG: Fix bugs and refactor ONNX output decoration

Samuel Van Stroud requested to merge svanstro/athena:svanstro/onnx-export-deco into main

Follow up to !68411 (merged)

Refactoring of the way we decorate outputs from the ONNX models back to the EDM.

  • Moved the function which builds the GNN decorators from DataPrepUtilities.cxx (which has too many responsibiltiies) to GNN.cxx. This improves cohesion.
  • (AFT-705) Fix a bug where taggers would overwrite the track links decoration which had a hardcoded name.
  • Fix a bug in the decoration of aux tasks back to the jet/btagging object, which will be useful for @jburzyns and @lmcelhin for https://its.cern.ch/jira/browse/ATLIDTRKCP-455 (bug introduced in !66718 (merged))

cc @ligang @bdong @dguest @dkobylia @dbiswas

Edited by Samuel Van Stroud

Merge request reports