diff --git a/analysis/tasks/FastProd.py b/analysis/tasks/FastProd.py index 7001d11497b4fa9eec58f4b47b1a1295b2d319e1..8e157b8c44923eb215b58076464a5f103e8d7774 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 }