Skip to content

Commit dfe864a

Browse files
committed
incorporate pieter's feedback
1 parent e354b04 commit dfe864a

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

openml/runs/functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ def _run_task_get_arffcontent(
401401
range(num_reps),
402402
range(num_folds),
403403
range(num_samples),
404-
)):
404+
), start=1):
405405

406406
train_indices, test_indices = task.get_train_test_split_indices(
407407
repeat=rep_no, fold=fold_no, sample=sample_no)
@@ -508,7 +508,7 @@ def _calculate_local_measure(sklearn_fn, openml_name):
508508
)
509509

510510
if len(traces) > 0:
511-
if len(traces) != n_fit + 1:
511+
if len(traces) != n_fit:
512512
raise ValueError(
513513
'Did not find enough traces (expected {}, found {})'.format(n_fit, len(traces))
514514
)

0 commit comments

Comments
 (0)