File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed
Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff 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+ )
You can’t perform that action at this time.
0 commit comments