ATLAS Offline Software
Loading...
Searching...
No Matches
python.CaloClusterTritonConfig Namespace Reference

Functions

 CaloClusterTritonToolCfg (flags, name="CaloClusterTritonTool", **kwargs)
 CaloClusterTritonAlgCfg (flags, clusterContainer="CaloCalTopoClusters", decorName="GNNCalibratedEnergy", name="CaloClusterTritonAlg", **kwargs)

Variables

 parser = argparse.ArgumentParser()
 nargs
 required
 True
 help
 default
 type
 int
 args = parser.parse_args()
 flags = initConfigFlags()
 Files
 MaxEvents
 cfg = MainServicesCfg(flags)

Function Documentation

◆ CaloClusterTritonAlgCfg()

CaloClusterTritonAlgCfg ( flags,
clusterContainer = "CaloCalTopoClusters",
decorName = "GNNCalibratedEnergy",
name = "CaloClusterTritonAlg",
** kwargs )
Schedule CaloClusterTritonAlg, which decorates clusters with the GNN energy [MeV].

Definition at line 23 of file CaloClusterTritonConfig.py.

27 **kwargs):
28 """Schedule CaloClusterTritonAlg, which decorates clusters with the GNN energy [MeV]."""
29 acc = ComponentAccumulator()
30
31 kwargs.setdefault("CaloClusterTritonTool",
32 acc.popToolsAndMerge(CaloClusterTritonToolCfg(flags)))
33 kwargs.setdefault("ClusterContainer", clusterContainer)
34 kwargs.setdefault("ClusterGNNCalibratedEnergyKeyName", f"{clusterContainer}.{decorName}")
35 kwargs.setdefault("RapidityRange", [-2.5, 2.5])
36 kwargs.setdefault("MinClusterEnergy", 300.0) # MeV
37
38 acc.addEventAlgo(CompFactory.CaloClusterTritonAlg(name, **kwargs))
39 return acc
40
41
42

◆ CaloClusterTritonToolCfg()

CaloClusterTritonToolCfg ( flags,
name = "CaloClusterTritonTool",
** kwargs )
Configure CaloClusterTritonTool with its AthInfer::TritonTool backend.

Definition at line 7 of file CaloClusterTritonConfig.py.

7def CaloClusterTritonToolCfg(flags, name="CaloClusterTritonTool", **kwargs):
8 """Configure CaloClusterTritonTool with its AthInfer::TritonTool backend."""
9 acc = ComponentAccumulator()
10
11 from AthTritonComps.TritonToolConfig import TritonToolCfg
12 kwargs.setdefault("TritonInferenceTool",
13 acc.popToolsAndMerge(TritonToolCfg(flags,
14 model_name=flags.Calo.Triton.name,
15 url=flags.Calo.Triton.url,
16 port=flags.Calo.Triton.port)))
17 kwargs.setdefault("CellEnergyThreshold", 0.005) # GeV
18
19 acc.setPrivateTools(CompFactory.CaloClusterTritonTool(name, **kwargs))
20 return acc
21
22

Variable Documentation

◆ args

python.CaloClusterTritonConfig.args = parser.parse_args()

Definition at line 55 of file CaloClusterTritonConfig.py.

◆ cfg

python.CaloClusterTritonConfig.cfg = MainServicesCfg(flags)

Definition at line 62 of file CaloClusterTritonConfig.py.

◆ default

python.CaloClusterTritonConfig.default

Definition at line 52 of file CaloClusterTritonConfig.py.

◆ Files

python.CaloClusterTritonConfig.Files

Definition at line 58 of file CaloClusterTritonConfig.py.

◆ flags

python.CaloClusterTritonConfig.flags = initConfigFlags()

Definition at line 57 of file CaloClusterTritonConfig.py.

◆ help

python.CaloClusterTritonConfig.help

Definition at line 50 of file CaloClusterTritonConfig.py.

◆ int

python.CaloClusterTritonConfig.int

Definition at line 53 of file CaloClusterTritonConfig.py.

◆ MaxEvents

python.CaloClusterTritonConfig.MaxEvents

Definition at line 59 of file CaloClusterTritonConfig.py.

◆ nargs

python.CaloClusterTritonConfig.nargs

Definition at line 50 of file CaloClusterTritonConfig.py.

◆ parser

python.CaloClusterTritonConfig.parser = argparse.ArgumentParser()

Definition at line 49 of file CaloClusterTritonConfig.py.

◆ required

python.CaloClusterTritonConfig.required

Definition at line 50 of file CaloClusterTritonConfig.py.

◆ True

python.CaloClusterTritonConfig.True

Definition at line 50 of file CaloClusterTritonConfig.py.

◆ type

python.CaloClusterTritonConfig.type

Definition at line 53 of file CaloClusterTritonConfig.py.