From 677954262ccb05657c7ca33833d0f255909e3f3c Mon Sep 17 00:00:00 2001
From: Klaus Rabbertz <klaus.rabbertz@cern.ch>
Date: Sun, 1 Oct 2023 17:19:08 +0200
Subject: [PATCH] Add multi_channel setting to FastProd

---
 analysis/tasks/FastProd.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/analysis/tasks/FastProd.py b/analysis/tasks/FastProd.py
index 7001d11..8e157b8 100644
--- a/analysis/tasks/FastProd.py
+++ b/analysis/tasks/FastProd.py
@@ -38,6 +38,7 @@ class FastProd(Task, TarballExtractionMixin, HTCondorWorkflow, law.LocalWorkflow
     starting_seeds = SpaceSeparatedListParameter(cls=luigi.IntParameter)
     fastprod_events = SpaceSeparatedListParameter(cls=luigi.IntParameter)
     fastprod_jobs = SpaceSeparatedListParameter(cls=luigi.IntParameter)
+    multi_channel = SpaceSeparatedListParameter(cls=luigi.Parameter)
 
     NNLOJET_RUNMODE   = 'production'
     NNLOJET_UNITPHASE = '! UNIT_PHASE'
@@ -52,7 +53,8 @@ class FastProd(Task, TarballExtractionMixin, HTCondorWorkflow, law.LocalWorkflow
                     'index': j,
                     'channel': channel,
                     'events': self.fastprod_events[j],
-                    'seed': str(k + int(self.starting_seeds[j]))
+                    'seed': str(k + int(self.starting_seeds[j])),
+                    'multi_channel': self.multi_channel[j],
                 }
                 i += 1
         return branchmap
@@ -158,6 +160,7 @@ class FastProd(Task, TarballExtractionMixin, HTCondorWorkflow, law.LocalWorkflow
                             'seed': self.branch_data['seed'],
                             'iterations': '1',
                             'events': self.branch_data['events'],
+                            'multi_channel': self.branch_data['multi_channel'],
                             'runmode': self.NNLOJET_RUNMODE,
                             'unit_phase': self.NNLOJET_UNITPHASE
                         }
-- 
GitLab