From 91e5abf48c5028f7043612b735bccc7e69fb71ce Mon Sep 17 00:00:00 2001 From: axif Date: Sat, 22 Nov 2025 22:54:22 +0600 Subject: [PATCH] Improve error message for inconsistent fold counts in OpenMLSplit --- openml/tasks/split.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/openml/tasks/split.py b/openml/tasks/split.py index 4e781df35..21c84ed95 100644 --- a/openml/tasks/split.py +++ b/openml/tasks/split.py @@ -56,9 +56,15 @@ def __init__( self.repeats = len(self.split) - # TODO(eddiebergman): Better error message - if any(len(self.split[0]) != len(self.split[i]) for i in range(self.repeats)): - raise ValueError("") + # Ensure all repetitions have the same number of folds + n_folds_rep_0 = len(self.split[0]) + for i in range(1, self.repeats): + if len(self.split[i]) != n_folds_rep_0: + raise ValueError( + f"Inconsistent number of folds across repetitions. " + f"Repetition 0 has {n_folds_rep_0} folds, " + f"but repetition {i} has {len(self.split[i])} folds.", + ) self.folds = len(self.split[0]) self.samples = len(self.split[0][0])