From 41c565c379892c9acefe3de15751486f8c663ed4 Mon Sep 17 00:00:00 2001
From: Thomas Strebler <thomas.strebler@cern.ch>
Date: Mon, 10 Jul 2023 15:22:22 +0200
Subject: [PATCH] Updated TRT_ElectronPidToolsConfig

Updated TRT_ElectronPidToolsConfig
---
 .../python/TRT_ElectronPidToolsConfig.py      | 23 ++++++++++++-------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/InnerDetector/InDetConfig/python/TRT_ElectronPidToolsConfig.py b/InnerDetector/InDetConfig/python/TRT_ElectronPidToolsConfig.py
index ae9bfa4d801e..9da2519e30da 100644
--- a/InnerDetector/InDetConfig/python/TRT_ElectronPidToolsConfig.py
+++ b/InnerDetector/InDetConfig/python/TRT_ElectronPidToolsConfig.py
@@ -60,26 +60,33 @@ def __TRT_dEdxToolBaseCfg(flags, name, **kwargs):
     acc = TRTToTCondAlgCfg(flags)
 
     kwargs.setdefault("TRT_dEdx_isData", not flags.Input.isMC)
-    
-    if "TRT_LocalOccupancyTool" not in kwargs:
-        kwargs.setdefault("TRT_LocalOccupancyTool", acc.popToolsAndMerge(TRT_LocalOccupancyCfg(flags)))
-
-    if "AssociationTool" not in kwargs:
-        from InDetConfig.InDetAssociationToolsConfig import InDetPrdAssociationToolCfg
-        kwargs.setdefault("AssociationTool", acc.popToolsAndMerge(InDetPrdAssociationToolCfg(flags)))
 
     acc.setPrivateTools(CompFactory.TRT_ToT_dEdx(name,**kwargs))
     return acc
 
+
 def TRT_dEdxToolCfg(flags, name="TRT_dEdxTool", **kwargs):
-    acc = __TRT_dEdxToolBaseCfg(flags, name, **kwargs)
+    acc = ComponentAccumulator()
     
     if not flags.Input.isMC:
         from LumiBlockComps.LumiBlockMuWriterConfig import LumiBlockMuWriterCfg
         acc.merge(LumiBlockMuWriterCfg(flags))
 
+    if "TRT_LocalOccupancyTool" not in kwargs:
+        kwargs.setdefault("TRT_LocalOccupancyTool", acc.popToolsAndMerge(
+            TRT_LocalOccupancyCfg(flags)))
+
+    if "AssociationTool" not in kwargs:
+        from InDetConfig.InDetAssociationToolsConfig import (
+            InDetPrdAssociationToolCfg)
+        kwargs.setdefault("AssociationTool", acc.popToolsAndMerge(
+            InDetPrdAssociationToolCfg(flags)))
+
+    acc.setPrivateTools(acc.popToolsAndMerge(
+        __TRT_dEdxToolBaseCfg(flags, name, **kwargs)))
     return acc
 
+
 def TrigTRT_dEdxToolCfg(flags, name="TrigTRT_dEdxTool", **kwargs):
     """trigger version should not add LumiBlockMuWriterCfg to views as it is scheduled globally"""
     
-- 
GitLab