Skip to content

Commit 0e59af6

Browse files
authored
[model cards] Prevent crash on generating widgets if dataset column is empty (#2997)
(or if it has no string columns)
1 parent 5e1a7a4 commit 0e59af6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

sentence_transformers/model_card.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -429,6 +429,9 @@ def set_widget_examples(self, dataset: Dataset | DatasetDict) -> None:
429429
]
430430
str_dataset = dataset[dataset_name].select_columns(columns)
431431
dataset_size = len(str_dataset)
432+
if dataset_size == 0:
433+
continue
434+
432435
lengths = {}
433436
for idx, sample in enumerate(
434437
str_dataset.select(random.sample(range(dataset_size), k=min(num_samples_to_check, dataset_size)))

0 commit comments

Comments
 (0)