Skip to content
Snippets Groups Projects

Update PLIT model for Run3 and add to PHYS

Merged Simone Francescato requested to merge sfrances/athena:run3plit into main
All threads resolved!
3 files
+ 21
12
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -299,20 +299,21 @@ def DecoratePLITCfg(
# path on calib area (found by path resolver
# /cvmfs/atlas.cern.ch/repo/sw/database/GroupData/
# Note: Run3 config is not available yet, so we use the Run2 config for now
kwargs.setdefault("ConfigPath", "IsolationSelection/PLIT/2025-02-24/")
if lepton_name == 'Electrons':
if isRun3:
kwargs.setdefault("ConfigFileVersion", 'network_electrons_barrel_run2.onnx')
kwargs.setdefault("ConfigFileVersion_endcap", 'network_electrons_endcap_run2.onnx')
kwargs.setdefault("ConfigPath", "IsolationSelection/PLIT/2025-03-24/")
kwargs.setdefault("ConfigFileVersion", 'network_electrons_barrel_run3.onnx')
kwargs.setdefault("ConfigFileVersion_endcap", 'network_electrons_endcap_run3.onnx')
else:
kwargs.setdefault("ConfigPath", "IsolationSelection/PLIT/2025-02-24/")
kwargs.setdefault("ConfigFileVersion", 'network_electrons_barrel_run2.onnx')
kwargs.setdefault("ConfigFileVersion_endcap", 'network_electrons_endcap_run2.onnx')
elif lepton_name == 'Muons':
if isRun3:
kwargs.setdefault("ConfigFileVersion", 'network_muons_run2.onnx')
kwargs.setdefault("ConfigPath", "IsolationSelection/PLIT/2025-03-24/")
kwargs.setdefault("ConfigFileVersion", 'network_muons_run3.onnx')
else:
kwargs.setdefault("ConfigPath", "IsolationSelection/PLIT/2025-02-24/")
kwargs.setdefault("ConfigFileVersion", 'network_muons_run2.onnx')
else:
raise ValueError(f'Decorate{Tagger_name} - unknown lepton type: "{lepton_name}"')
Loading