diff --git a/columnflow/tasks/cutflow.py b/columnflow/tasks/cutflow.py index e18fbbdbe..2337d6a57 100644 --- a/columnflow/tasks/cutflow.py +++ b/columnflow/tasks/cutflow.py @@ -102,10 +102,6 @@ def __init__(self, *args, **kwargs): "there, 'raise' is assumed", ) - def create_branch_map(self): - # dummy branch map - return [None] - def workflow_requires(self): reqs = super().workflow_requires() reqs["lfns"] = self.reqs.GetDatasetLFNs.req(self) @@ -210,7 +206,7 @@ def expr(events): expressions[variable_inst.name] = expr # prepare columns to load - load_columns = read_columns | set(mandatory_coffea_columns) + load_columns = read_columns | {Route(c) for c in mandatory_coffea_columns} load_sel_columns = {Route("steps.*")} # prepare histograms