3 from AthenaConfiguration.ComponentAccumulator
import ComponentAccumulator
4 from AthenaConfiguration.ComponentFactory
import CompFactory
8 the_alg = CompFactory.MuonML.InferenceAlg(name, **kwargs)
9 result.addEventAlgo(the_alg, primary =
True)
14 from AthOnnxComps.OnnxRuntimeSessionConfig
import OnnxRuntimeSessionToolCfg
18 kwargs.setdefault(
"ModelSession", result.popToolsAndMerge(
OnnxRuntimeSessionToolCfg(flags, model_fname=
"/eos/atlas/atlascerngroupdisk/data-art/grid-input/MuonRecRTT/TestModel.onnx")))
19 kwargs.setdefault(
"MLFilterCut", -2.7)
21 the_tool = CompFactory.MuonML.GraphBucketFilterTool(name, **kwargs)
22 result.setPrivateTools(the_tool)