fix requires from MLEvaluationSingleFold

uhh-cms · Aug 12, 2024 · e4ff078 · e4ff078
1 parent 3ddaa2d
commit e4ff078
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/hbw/tasks/ml.py b/hbw/tasks/ml.py
@@ -437,6 +437,7 @@ def requires(self):
         reqs["training"] = self.reqs.MLTraining.req_different_branching(
             self,
             branches=(self.fold,),
+            branch=self.fold,
             configs=(self.config_inst.name,),
             calibrators=(self.calibrators,),
             selectors=(self.selector,),