Skip to content

Commit ed1af7e

Browse files
committed
make lint
1 parent cb3893b commit ed1af7e

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

mteb/abstasks/AbsTaskClusteringFast.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -320,9 +320,9 @@ def convert_to_fast(
320320

321321
# check that it is the same distribution
322322
row_label_set = set(lab)
323-
assert row_label_set.issubset(
324-
all_labels_set
325-
), "The clusters are not sampled from the same distribution as they have different labels."
323+
assert row_label_set.issubset(all_labels_set), (
324+
"The clusters are not sampled from the same distribution as they have different labels."
325+
)
326326

327327
for l, s in zip(lab, sents):
328328
if s not in sent_set:
@@ -353,6 +353,6 @@ def check_label_distribution(ds: DatasetDict) -> None:
353353

354354
# check that it is the same distribution
355355
row_label_set = set(lab)
356-
assert row_label_set.issubset(
357-
all_labels_set
358-
), "The clusters are not sampled from the same distribution as they have different labels."
356+
assert row_label_set.issubset(all_labels_set), (
357+
"The clusters are not sampled from the same distribution as they have different labels."
358+
)

0 commit comments

Comments
 (0)