From cc28b60365f276e21c8f7ef601ac2ac47b6682e4 Mon Sep 17 00:00:00 2001 From: Robinpreet Dhillon <rdhill13@narval2.narval.calcul.quebec> Date: Wed, 30 Mar 2022 15:26:56 -0400 Subject: [PATCH] added shaper function --- .../imaging_shower_ML/options/imaging_ml_data.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py index 4b8188c9..fbdcf932 100644 --- a/benchmarks/imaging_shower_ML/options/imaging_ml_data.py +++ b/benchmarks/imaging_shower_ML/options/imaging_ml_data.py @@ -12,7 +12,7 @@ from Configurables import Jug__Reco__CalorimeterHitsMerger as CalHitsMerger from Configurables import Jug__Reco__ImagingPixelMerger as MLDataMerger from Configurables import Jug__Reco__ImagingPixelDataSorter as MLDataSorter from Configurables import Jug__Reco__ImagingPixelDataCombiner as MLDataCombiner - +from Configurables import Jug__Reco__ImagingPixelDataShaper as MLDataShaper # input arguments through environment variables kwargs = dict() @@ -123,7 +123,11 @@ becal_combiner = MLDataCombiner('becal_combiner', layerIncrement=100, rule=kwargs['combine']) - + +# shape layers +becal_shaper = MLDataShaper('becal_combiner', + inputHits=becal_combiner.outputHits) + podout.outputCommands = [ # 'keep *', @@ -138,7 +142,7 @@ ApplicationMgr( TopAlg=[podin, becal_img_digi, becal_img_reco, becal_img_merger, becal_img_sorter, becal_scfi_digi, becal_scfi_reco, becal_scfi_merger, becal_scfi_sorter, - becal_combiner, + becal_combiner, becal_shaper, podout], EvtSel='NONE', EvtMax=kwargs['nev'], -- GitLab