Skip to content

Commit f43f36e

Browse files
committed
fix regression task
1 parent c96c0c4 commit f43f36e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mteb/abstasks/regression.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ class AbsTaskRegression(AbsTaskClassification):
100100

101101
def _undersample_data(
102102
self, dataset: Dataset, experiment_num: int, idxs: list[int] | None = None
103-
) -> tuple[Dataset, list[int]]:
103+
) -> tuple[Dataset, list[int], list[int]]:
104104
if self.n_samples >= len(dataset):
105105
train_split_sampled = dataset
106106
else:
@@ -111,7 +111,7 @@ def _undersample_data(
111111
label=self.label_column_name,
112112
n_samples=self.n_samples,
113113
)["train"]
114-
return train_split_sampled, []
114+
return train_split_sampled, [], []
115115

116116
def _calculate_scores(
117117
self,

0 commit comments

Comments
 (0)